【数据结构】堆的初始化——如何初始化一个大根堆?

文章目录

  • 源码是如何插入的?
  • 扩容
  • 向上调整
  • 实现大根堆
    • 代码:

源码是如何插入的?

【数据结构】堆的初始化——如何初始化一个大根堆?_第1张图片

扩容

【数据结构】堆的初始化——如何初始化一个大根堆?_第2张图片

在扩容的时候,如果容量小于64,那就2倍多2的扩容;如果大于64,那就1.5倍扩容。

还会进行溢出的判断,如果决定的新容量大于给的数组最大容量,那就将其限制在数组最大容量:
【数据结构】堆的初始化——如何初始化一个大根堆?_第3张图片

向上调整

【数据结构】堆的初始化——如何初始化一个大根堆?_第4张图片

在进行向上调整的时候,会对传进来的comparator进行判断,如果不为空,那就使用程序员传进来的比较器接口,如果为空,那就说明调用者并未实现比较器,那么就使用java自己提供的函数siftUpComparable(k, x);

传进来的x就是要插入的值e。

【数据结构】堆的初始化——如何初始化一个大根堆?_第5张图片

这是使用程序员自己传进来的比较器进行比较,调用了compare接口进行比较,所以要想初始化一个大根堆,那就得自己写出一个compare函数然后传进去。

【数据结构】堆的初始化——如何初始化一个大根堆?_第6张图片

在使用自己写的compare函数时,会让x强转为Comparable类型,如果这个x不是可以比较的(未实现Comparable接口,那就会抛出类型转换异常)

实现大根堆

值得说明的是:
比较器函数是Comparator而不是Comparable。

代码:

import java.util.Comparator;
import java.util.PriorityQueue;

class IntCmp implements Comparator<Integer> {

    @Override
    public int compare(Integer o1, Integer o2) {
        // 当然,写o1.compareTo(o2)仍然是小根堆
        return o2.compareTo(o1);
    }
}

public class Test {
    public static void main(String[] args) {
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>();
        priorityQueue.offer(1);
        priorityQueue.offer(2);
        priorityQueue.offer(3);
        priorityQueue.offer(4);

        System.out.print(priorityQueue.poll());
        System.out.print(priorityQueue.poll());
        System.out.print(priorityQueue.poll());
        System.out.println(priorityQueue.poll());// 1 2 3 4,可以发现java中提供的默认堆是小根堆

        System.out.println("======");

        PriorityQueue<Integer> priorityQueue1 = new PriorityQueue<>(new IntCmp());
        priorityQueue1.offer(1);
        priorityQueue1.offer(2);
        priorityQueue1.offer(3);
        priorityQueue1.offer(4);

        System.out.print(priorityQueue1.poll());
        System.out.print(priorityQueue1.poll());
        System.out.print(priorityQueue1.poll());
        System.out.print(priorityQueue1.poll());// 4 3 2 1,变为大根堆
    }
}

你可能感兴趣的:(数据结构,数据结构)