二叉堆
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里很多类型都是不可变的。
这有很多好处,但是坏处就是如果你需要改变值必须要新建一个对象。