堆排序

堆,大根堆,小根堆

堆是一棵完全二叉树并且分为大根堆和小根堆。大根堆为其任意非叶子结点都不小于其孩子结点,小根堆则为其任意非叶子结点都不大于其孩子结点。
这里用一组数分别构建大根堆与小根堆。
此为大根堆:
大根堆
此为小根堆:
小根堆
堆排序的思路如下:
步骤一:将n个元素构成的序列构建成大根堆。
步骤二:将最后一个元素与堆顶元素互换,这时堆尾元素为最大元素。
步骤三:将排除堆尾元素的剩下n-1个元素继续构建大根堆。
然后重复这几个步骤直到序列有序。

构建大根堆

现有如下数组其数据元素为 5,7,6,2,8
在这里插入图片描述

如果按照第i个元素的第2i + 1个元素是其左孩子,第2i + 2个元素是其右孩子,我们可以构建出一颗完全二叉树。
在这里插入图片描述

在二叉树中0号元素的左孩子为1号,右孩子为2号,符合上面提到的特点。
然后我们从最后一个非叶结点来调整,然后我们发现7要比8小,所以交换7和8的位置后:
在这里插入图片描述
然后继续查看下一个非叶子结点,发现6其实是叶子结点因为没有左右孩子,所以下一个非叶子结点是5,对比5和它的左右孩子,将最大的8和5交换
在这里插入图片描述
但这次交换之后破坏了根节点左子树的大根堆,所以要继续对5进行调整
在这里插入图片描述
这样现在的完全二叉树就是一个大根堆了,构建出大根堆之后,我们把堆顶元素和最后一个结点交换位置。
在这里插入图片描述
此时数组就变成了这样,最后一个元素一定是最大的,第一轮结束后,将序列的个数减一然后继续调整大根堆最后再与之交换,也就是5, 7, 6, 2这四个元素调整为大根堆。
为了用代码实现堆排序,我们还要知道完全二叉树的最后一个非叶结点的下标为size / 2 - 1。首先根节点下标为0,总共有size个元素,那么最后一个节点的下标就是size - 1,而且我们知道一个节点如果下标是i,那么它的左孩子下标就是2 * i + 1,右孩子下标就是2 * i + 2,现在有以下两种情况:

  1. 最后一个非叶节点只有左孩子:
    最后一个节点下标是size - 1,且是左孩子所以有2*i + 1 = size - 1化简得到i = size / 2 - 1。
  2. 最后一个非叶节点有左右孩子:
    最后一个节点下标是size - 1,且是右孩子所以有2*i + 2 = size - 1向下取整化简是i = size / 2 - 1。

堆排序代码

//构建大根堆
//传入数组,数组长度以及最后一个非叶结点的下标
void buildHeap(vector<int>& nums, int size, int index) {
	int lchild = 2 * index + 1;
	int rchild = 2 * index + 2;
	int max = index;
	if (lchild < size && nums[max] < nums[lchild]) {
		max = lchild;
	}
	if (rchild < size && nums[max] < nums[rchild]) {
		max = rchild;
	}
	//当最大的元素不是当前非叶结点时交换
	if (max != index) {
		swap(nums[max], nums[index]);
		buildHeap(nums, size, max);
	}

}

//堆排序
void heapSort(vector<int>& nums, int size) {
	//从最后一个非叶结点开始构建大根堆
	for (int i = size / 2 - 1; i >= 0; --i) {
		buildHeap(nums, size, i);
	}
	// 调整大根堆
	for (int i = size - 1; i >= 1; i--)
	{
		swap(nums[0], nums[i]);           // 将当前最大的放置到数组末尾
		buildHeap(nums, i, 0);            // 将未完成排序的部分继续进行堆排序
	}
}

int main(int argc, char* argv[]) {
	vector<int> ival;
	int num = 0;
	while (cin >> num) {
		ival.push_back(num);
	}
	heapSort(ival, ival.size());
	for (auto i : ival) {
		cout << i << endl;
	}
	return 0;
}

平均时间复杂度:O(n*logn)
平均空间复杂度:O(1)
稳定性:不稳定

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值