C++实现堆排序

堆排序

什么是堆

堆是一种叫做完全二叉树的数据结构,可以分为大顶堆和小顶堆。

堆的分类

大顶堆:每个结点的值都大于或者等于他的左右结点的值

小顶堆:每个结点的值都小于或者等于他的左右结点的值

下标为 i 的父节点下标: (i - 1) / 2
下标为 i 的左孩子下标: i * 2 + 1
下标为 i 的右孩子下标: i * 2 + 2

排序思想

1、首先将待排序的数组构造成一个大根堆,此时,整个数组的最大值就是堆结构的顶端

2、将顶端的数与末尾的数交换,此时,末尾的数为最大值,剩余待排序数组个数为n - 1

3、将剩余的n - 1个数再构造成大根堆,再将顶端数与n -1位置交换,如此反复执行,便能得到有序数组

注意:升序用大根堆,降序用小根堆
因为大根堆每次都将最大的数交换到数组末尾
小顶堆每次都将最小的数交换到数组末尾

维护堆

如果二叉树中的某个结点不满足堆的性质,就要对其进行维护

以大顶堆为例:

如果发现一个结点比他的左孩子或者右孩子小,就将该节点与值最大的孩子结点进行交换

cpp 复制代码
/*
维护堆
	num:表示存储堆的数组
	n 数组长度
	i 待维护结点的下标
*/
void hepify(vector<int>& nums, int n, int i) {
	int largest = i;
	int lson = i * 2 + 1;
	int rson = i * 2 + 2;
	if (lson < n && nums[largest] < nums[lson])
		largest = lson;
	if (rson < n && nums[largest] < nums[rson])
		largest = rson;
	//有个孩子比该结点大
	if (largest != i) {
		swap(nums[i], nums[largest]);
		//调整后,下面的结点可能依旧不满足,要递归执行
		hepify(nums, n, largest);
	}
}

构建堆

1、从最后一个子树开始,从后往前调整

2、每次调整,从上往下调整

3、调整为大根堆

首先将给定的一个无序队列看成一个堆结构,一个没有规则的二叉树,将序列里的值按照从上往下,从左到右依次填充到二叉树中

如何构建堆呢? 找到最后一个非叶子结点,也就是下标为arr.size() / 2 - 1的结点,比较它的左右节点中最大的一个的值,是否比他大,如果大就交换位置。


此时就构建出来一个大顶堆。

接下来进行排序


因此可以归纳出堆排序的步骤:

1、将无序数组构建成堆

2、循环删除堆顶元素,移动集合尾部,调节堆产生新的堆顶

cpp 复制代码
void heap_sort(vector<int>& nums, int n) {
	//建堆
	//i = n / 2 - 1是最后一个非叶子结点的下标
	//依次调整非叶子结点的下标,直到堆顶为止
	for (int i = n / 2 - 1; i >= 0; i--) {
		hepify(nums, n, i);
	}

	//排序
	for (int i = n - 1; i > 0; i--) {
		swap(nums[i], nums[0]);
		hepify(nums, i, 0);
	}

}

完整代码

cpp 复制代码
/*
维护堆
	num:表示存储堆的数组
	n 数组长度
	i 待维护结点的下标
*/
void hepify(vector<int>& nums, int n, int i) {
	int largest = i;
	int lson = i * 2 + 1;
	int rson = i * 2 + 2;
	if (lson < n && nums[largest] < nums[lson])
		largest = lson;
	if (rson < n && nums[largest] < nums[rson])
		largest = rson;
	//有个孩子比该结点大
	if (largest != i) {
		swap(nums[i], nums[largest]);
		//调整后,下面的结点可能依旧不满足,要递归执行
		hepify(nums, n, largest);
	}
}

void heap_sort(vector<int>& nums, int n) {
	//建堆
	//i = n / 2 - 1是最后一个非叶子结点的下标
	//依次调整非叶子结点的下标,直到堆顶为止
	for (int i = n / 2 - 1; i >= 0; i--) {
		hepify(nums, n, i);
	}

	//排序
	for (int i = n - 1; i > 0; i--) {
		swap(nums[i], nums[0]);
		hepify(nums, i, 0);
	}

}

int main() {
	vector<int> nums{ 30, 24, 5, 58, 18, 36, 12, 42, 39 };
	cout << "排序前:";
	for (int num : nums)
		cout << num << " ";
	cout << endl;

	heap_sort(nums, nums.size());
	cout << "堆排序后:";
	for (int num : nums)
		cout << num << " ";
	cout << endl;
}

结果:
排序前:30 24 5 58 18 36 12 42 39
堆排序后:5 12 18 24 30 36 39 42 58
相关推荐
GalaxyPokemon5 小时前
归并排序:分治思想的高效排序
数据结构·算法·排序算法
曼巴UE58 小时前
UE5 音效系统
c++·游戏·ue5·虚幻·音效
无影无踪的青蛙9 小时前
[C++] list双向链表使用方法
c++·链表·list
tkevinjd10 小时前
C++中线程库的基本操作
开发语言·c++
CodeWithMe10 小时前
【C/C++】不同防止头文件重复包含的措施
c语言·开发语言·c++
Fre丸子_10 小时前
C++定长内存块的实现
c++
子豪-中国机器人10 小时前
C++ 信息学奥赛总复习题答案解析
开发语言·c++·算法
弥彦_11 小时前
牛客round95D
c++·算法
oioihoii11 小时前
C++11列表初始化:从入门到精通
java·开发语言·c++
tomato0911 小时前
2025 年中国大学生程序设计竞赛全国邀请赛(郑州)暨第七届CCPC河南省大学生程序设计竞赛(补题)
c++