C++排序 之 堆排序

一、理论知识

以升序为例，堆排序简单来说就是先将原数组调整为大堆，此时根节点的数最大，最后一个叶子节点数最小，所以需将这两个节点的值进行交换，然后保证最后这个叶子节点的数不变，并从根节点开始往下调整，使除最后一个叶子节点之外的数维持大堆状态。同理，继续将调整后的根节点与倒数第二个叶子节点进行交换，然后完成以上操作，以此类推，直至调整到根节点。

具体堆排序的理论只是可见博主的博客 温故而知新 -＞ 数据结构 -＞排序。
同时，因为堆排序基于二叉树的顺序结构存储，因此在代码编写中会利用二叉树的一些性质，若有不懂可见博主的博客 二叉树介绍 ~ 概念、存储结构、性质。

二、程序与结果

2.1 C++代码

#include <iostream>
#include <vector>
using namespace std;

void display(vector<int>& arr)
{
	for (auto& n : arr)
		cout << n << " ";
	cout << endl;
}

void _heapSort(vector<int>& arr, int len, int idx)
{
	int left = 2 * idx + 1;//左子节点的索引
	int right = 2 * idx + 2;//右子节点的索引
	int maxidx = idx;//用于保存idx指向的节点与两个子节点中最大值的索引
	if (left < len && arr[left] > arr[maxidx])
		maxidx = left;
	if (right < len && arr[right] > arr[maxidx])
		maxidx = right;
	if (maxidx != idx)
	{
		swap(arr[maxidx], arr[idx]);//将最大值置于根节点处
		_heapSort(arr, len, maxidx);//从交换过的节点的位置继续向下调整
	}
}

void heapSort(vector<int>& arr)
{
	int len = arr.size();
	//先将原数组进行大堆调整
	for (int i = len / 2 - 1; i >= 0; i--)
		_heapSort(arr, len, i);
	//对调整后的数组进行调整
	for (int i = len - 1; i >= 1; i--)//此处判断条件 i >= 1 也可设置为 i >= 0，只不过如此会产生一些无用的工作量
	{
		swap(arr[0], arr[i]);
		_heapSort(arr, i, 0);
	}
}

void test()
{
	vector<int> arr = { 5, 2, 8, 6, 4, 7, 9, 1, 3 };
	cout << "原数组：";
	display(arr);
	heapSort(arr);
	cout << "堆排序后：";
	display(arr);
}

int main()
{
	test();
	return 0;
}

2.2 运行结果