堆粗浅的总结

堆的特性:是一个完全二叉树,假设下标从零开始,对于索引cur而言:

  • 左孩子:2*cur+1;
  • 右孩子:2*cur+2;
  • 父节点:(cur-1)/2

最大堆排序:参考1 参考2

void maxHeapDown(vector<int> &nums, int start, int end) {
	int left = start * 2 + 1;
	int right = start * 2 + 2;
	int largeIndex;
	while (left <= end) {
		if (right <= end && nums[left] < nums[right]) {
			largeIndex = right;
		}
		else {
			largeIndex = left;
		}
		if (nums[start] > nums[largeIndex]) {
			largeIndex = start;
		}
		if (largeIndex == start) {
			break;
		}
		swap(nums[start], nums[largeIndex]);
		start = largeIndex;
		left = 2 * start + 1;
		right = 2 * start + 2;
	}
}
void maxHeapDown2(vector<int> &nums, int start, int end) {
	int c = start;
	int l = 2 * c + 1;
	int tmp = nums[c];
	for (; l <= end; c = l, l = 2 * c + 1) {
		if (l < end&&nums[l] < nums[l + 1]) {
			l++;
		}
		if (tmp >= nums[l]) {
			break;
		}
		else {
			nums[c] = nums[l];
			nums[l] = tmp;
			
		}
	}
}
void buildMaxHeap(vector<int> &nums) {
	for (int i = nums.size() / 2 - 1; i >= 0; i--) {  //倒数第二层从右往左第一个子树
		maxHeapDown(nums, i, nums.size() - 1);  //从低向上
	}
}
void MaxHeapSortAsc(vector<int> &nums) {
	buildMaxHeap(nums);
	for (int i = nums.size() - 1; i >= 0; i--) {
		swap(nums[i], nums[0]);
		maxHeapDown(nums, 0, i - 1);
	}
}

int findKthLargest(vector<int>& nums, int k) {
	buildMaxHeap(nums);
	for (int i = nums.size() - 1; i >= nums.size() - k + 1; i--) {
		swap(nums[0], nums[i]);
		maxHeapDown(nums, 0, i - 1);
	}
	return nums[0];
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值