当时在学堆排序的时候误以为只需要判断父节点,左子树,右子树的大小然后再交换位置即完成了建堆,但是我忽略了一点那就是从底层开始确实是对的,但是越往上层,你不能够保证父节点会大于他的子树的所有元素所以需要对最大元素编号的子树再建一次堆(追加判断!!!)
递归版本
void HeapAdjust(int i,int size) {
int lchild = 2 * i, int rchild = 2 * i + 1;
int max = i;
if(i <= size / 2) { // 确保是否为父节点
if(lchild <= size && nums[lchild] > nums[max]) {
max = lchild;
}
if(rchild <= size && nums[rchild] > nums[max]) {
max = rchild;
}
if(max != i) {
swap(&nums[i], &nums[max]);
HeapAdjust(max, size); //避免调整之后以max为父节点的子树不是堆
}
}
}
迭代版本
void HeapAdjust(int i,int size) {
int l = i * 2, r = i * 2 + 1;
bool swapped = false;
int s = l;
while(!swapped && i <= size / 2) {
l = i * 2, r = i * 2 + 1, s = l;
// 右子树存在, 且大于左子树
if(r <= size && nums[l] < nums[r]) {
s = r;
}
//已经大于两边可直接退出
if(nums[s] < nums[i]) {
swapped = true;
} else {// 需要交换的话一定是父节点小于左右子树,
//在两个孩子节点的下面都是建好堆了的,
//你能保证交换过来的父节点一定会大于子树所有元素吗?
swap(&nums[s], &nums[i]);
i = s;
}
}
}