可以通过siftdown或者siftup建立二叉堆。
- siftdown: 如果节点小于其最大的子节点,将二者进行交换,直到节点大于或等于所有子节点。整个过程节点向下调整。
- siftup: 如果节点大于其父节点,将二者交换,直到节点小于或者等于其父节点。整个过程节点向上调整。
siftdown或者siftup在最坏情况下都是O(logn)。siftdown操作次数取决于节点离堆底的距离,siftup的操作次数取决于节点离堆顶的距离。siftdown的效率更高一些,因为建堆时计算每个节点到底部或者顶部的距离,由于大部分节点靠近底部,siftdown的总体操作次数少一些。
通过siftdown建堆
代码
//swap current value with the largest child.
private static void siftdown(int[] arr, int k, int len){
while(2 * k + 1 < len){
int child = 2 * k + 1;
if(2 * k + 2 < len && arr[2 * k + 1] < arr[2 * k + 2]){
child = 2 * k + 2;
}
if(arr[k] >= arr[child]){
break;
}
int temp = arr[k];
arr[k] = arr[child];
arr[child] = temp;
k = child;
}
}
//O(logn)
public static void buildheap(int[] arr){
for(int i = arr.length - 2; i >= 0; i--){
siftdown(arr, i, arr.length);
}
}
二叉堆中
倒数第一层 (堆底)的节点不需要通过siftdown交换,0次操作 * ( n/2 个节点 )
倒数第二层的节点最多进行1次交换到堆底,1次操作 * n/4个节点
倒数第三层的节点最多进行2次交换到堆底,2次操作 * n/8个节点
…
第一层的最多进行h (h 是二叉堆的深度)次交换到堆底,h次操作 * 1 个节点
如何确定每一层的节点数?
假设二叉堆的最后一层是满的。二叉树所有节点数和为
1 + 2 + 4 + . . . + 2 h = n 2 h + 1 − 1 = n 2 h = n + 1 2 h = l o g 2 ( n + 1 ) − 1 1 + 2 + 4 + ... + 2^h = n \\ 2^{h + 1} - 1 = n \\ 2^h = \frac{n + 1}{2} \\ h = log_2(n + 1) - 1 1+2+4+..