思想
利用一个一维数组通过下标来实现一个大根堆,大根堆其实就是一棵完全二叉树
对于任意一个位置 i 上的节点,他的父节点下标为 (i-1)/2
他的左孩子下标为 2i+1 , 他的右孩子下标为 2i+2
对于0位置上的节点,(0-1)/2就是0,因此0位置上的节点的根节点就是他自己
大根堆就是要保证每一棵子树的根节点是这棵树上最大的
建立大根堆
时间复杂度O(N)
void swap(int arr[], int i, int j){
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
void bigRootHeap(int arr[], int size){
for(int i = 0; i < size; i++){
for(int j = i; j >= 0; j = (j-1)/2){
if(arr[j] > arr[(j-1)/2]){
swap(arr, j, (j-1)/2);
}else{
break;
}
}
}
}
堆的调整(heapify)
我们需要在已经建立好的堆中有一个节点发生变化,就需要对原来已经构建好的堆进行调整
如果当前节点变小了,那么就需要找到他的两个孩子,然后与两个孩子中最大的那个节点进行比较,如果当前更新的节点比最大的孩子小则与最大的孩子进行交换
// heapSize 堆的大小
void heapify(int arr[], int loc, int newNum, int heapSize){
arr[loc] = newNum;
int L = 2*loc+1; // 左孩子
while(L < heapSize){
int bigLoc = L+1 < heapSize && arr[L+1] > arr[L] ? L+1 : L; //最大的孩子的下标
if(arr[bigLoc] > arr[loc]){
swap(arr, loc, bigLoc);
}else{
break;
}
loc = bigLoc;
L = 2*loc+1;
}
}