《Algorithms算法》笔记：优先队列(2)——二叉堆

二叉堆

1 二叉堆的定义

堆是一个完全二叉树结构（除了最底下一层，其他层全是完全平衡的），如果每个结点都大于它的两个孩子，那么这个堆是有序的。

二叉堆是一组能够用堆有序的完全二叉树排序的元素，并在数组中按照层级存储（不用数组的第一个位置）

 

2 二叉堆的性质

• 最大的元素在a[1] (root结点)
• 每个k的父亲在k/2
• 每个k的孩子在k*2和k*2+1

3 二叉堆的操作

3.1 上浮（孩子大于父亲）——对应插入操作

循环，每次比较自己和父亲，如果比父亲大就交换，直到root。

 

3.2 插入

先把元素加入数组的最后一个位置，然后进行上浮到正确位置

 

3.3 下沉（父亲小于儿子）——对应删除操作

循环，每次比较自己和两个孩子，如果比孩子小就交换，如果比两个孩子都小，就交换到两个里面较大的一个。直到叶子。

 

3.4 删除最大元素

先把根元素与最后一个元素交换，删除最后一个元素，然后从根开始下沉到正确位置。

 

4 二叉堆优先队列代码

/**
 *
 * @author rocky
 */
public class MaxPQ<Key extends Comparable<Key>> {

    private Key[] pq;
    private int N;

    public MaxPQ(int capacity) {
        pq = (Key[]) new Comparable[capacity + 1];
    }

    public boolean isEmpty() {
        return N == 0;
    }

    public void insert(Key key)
    {      
        N++; 
        pq[N] = key;
        swim(N);
    }

    public Key delMax() {
        Key max = pq[1];  //get the max element
        exch(1, N);     //exchange between the root and the last element
        N--;   
        sink(1);   //sink to the right place
        pq[N+1] = null;   //delete
        return max;
    }

    private void swim(int k)
    {
        while(k > 1 && less(k/2, k))
        {
            exch(k, k/2);
            k /= 2;
        }

    }

    private void sink(int k) {
        while(k*2 <= N)    //if this node has left child 
        {
            int child = k * 2;   
            if (child < N && less(child, child + 1)) {  //if the left child is less than the right child
                child = child + 1;   //change child to the right child
            }
            if (less(k, child)) {
                exch(k, child);
            }
            k = child;
        }
    }

    private boolean less(int i, int j) {
        return pq[i].compareTo(pq[j]) < 0;
    }

    private void exch(int i, int j) {
        Key t = pq[i];
        pq[i] = pq[j];
        pq[j] = t;
    }
}

5 二叉堆扩展

 

 

5.1 不可变性

我们算法的前提是用户不会改变队列的元素，如果用户能改变队列的元素，那么队列成立的条件就会破坏，
不可变的数据类型，就是说一个实例一旦建立，就不可以改变。java里很多类型都是不可变的。

 

这有很多好处，但是坏处就是如果你需要改变值必须要新建一个对象。