堆向下调整的实现

我们给出一个数组,逻辑上看做一颗完全二叉树。我们通过从根节点开始的向下调整算法可以把它调整成一个小堆。向下调整算法有一个前提:左右子树必须是一个堆,才能调整。


思路

以小堆为例,向下调整的实质就是将根节点与比它小的节点交换,逐渐下移,直到其子节点不再比它小为止。由此我们可以进行以下操作:

1、将根节点设为cur
2、进入循环,利用2 * cur + 1 找到左孩子,比较左孩子和右孩子的值,取其中较小的一个child
3、cur与这个child进行比较。若cur > child,则交换两数,并且将cur的位置更新为child,child的位置更新为2 * cur + 1;若cur <= child,则结束循环。注意,当child的位置越界时,也会结束循环。

代码实现

void shiftDown(int* arr, int n, int cur){
	int child = 2 * cur + 1;
	
	while (child < n){
		if (child + 1 < n && arr[child] > arr[child + 1]){
			child++;
		}
		if (arr[child] < arr[cur]){
			int tmp = arr[cur];
			arr[cur] = arr[child];
			arr[child] = tmp;

			cur = child;
			child = 2 * cur + 1;
		}
		else{
			break;
		}
	}
}

void test(){
	int arr[] = { 10, 5, 3, 8, 7, 6 };
	shiftDown(arr, sizeof(arr) / sizeof(arr[0]), 0);
}

void main(){
	test();
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值