堆排序的基本思路、操作、代码

关于堆排序，网上有很多讲解，在学习该算法时我曾找到一个比较好懂的链接（图解推排序），与大家一起分享。

堆排序的基本思路

1. 将无序序列构建成一个堆，根据升序降序需求选择大顶堆或小顶堆(一般升序采用大顶堆，降序采用小顶堆);
2. 将堆顶元素与末尾元素交换，将最大元素"沉"到数组末端;
3. 重新调整结构，使其满足堆定义，然后继续交换堆顶元素与当前末尾元素，反复执行调整+交换步骤，直到整个序列有序。

堆的操作

在堆的数据结构中，堆中的最大值总是位于根节点（在优先队列中使用堆的话堆中的最小值位于根节点）。堆中定义以下几种操作：

• 最大堆调整（Max Heapify）：将堆的末端子节点作调整，使得子节点永远小于父节点
• 创建最大堆（Build Max Heap）：将堆中的所有数据重新排序
• 堆排序（HeapSort）：移除位在第一个数据的根节点，并做最大堆调整的递归运算

代码

#include <iostream>
#include <algorithm>

using namespace std;

const int maxn = 100;
//使用数组来存储完全二叉树，1号位存储第一个节点
int heap[maxn], n;//heap为堆，n为元素个数

//对heap数组在[low, high]范围进行向下调整
//其中low为欲调整节点的数组下标，high一般为堆的最后一个元素的数组下标
void downAdjust(int low, int high) {
	int i = low, j = i * 2;//i为欲调整节点，j为其左孩子
	while (j <= high) {
		//如果有孩子存在且右孩子的值大于左孩子
		if (j + 1 <= high && heap[j + 1] > heap[j]) {
			j++;//让j存储右孩子下标
		}
		//如果孩子中最大的权值比欲调整节点i大
		if (heap[j] > heap[i]) {
			swap(heap[j], heap[i]);//交换最大权值的孩子与欲调整节点i
			i = j;
			j = j * 2;
		}
		else {
			break;
		}
	}
}

//建堆
void createHeap() {
	for (int i = n / 2; i >= 1; i--) {
		downAdjust(i, n);
	}
}

//删除堆顶元素
void deleteTop() {
	heap[1] = heap[n--];//用最后一个元素覆盖堆顶元素，并让元素个数减1
	downAdjust(1, n);//向下调整堆顶元素
}

//对heap数组在[low, high]范围进行向上调整
//其中low一般设置为1，high表示欲调整节点的数组下标
void upAdjust(int low, int high) {
	int i = high, j = i / 2;//i为欲调整节点，j为其父亲
	while (j >= low) {//父亲在[low, high]范围内
		//父亲权值小于欲调整节点i的权值
		if (heap[j] < heap[i]) {
			swap(heap[j], heap[i]);//交换父亲和欲调整节点
			i = j;//保持i为欲调整节点，j为i的父亲
			j = i / 2;
		}
		else {
			break;//父亲权值比欲调整节点i的权值大，调整结束
		}
	}
}

//添加元素x
void insert(int x) {
	heap[++n] = x;//让元素个数加1，然后将数组末位赋值为x
	upAdjust(1, n);//向上调整新加入的结点n
}

//堆排序
void heapSort() {
	createHeap();//建堆
	for (int i = n; i > 1; i--) {//倒着枚举，直到堆中只有一个元素
		swap(heap[i], heap[1]);//交换heap[i]与堆顶
		downAdjust(1,i-1);//调整堆顶
	}
}

int main() {
	n = 10;//令n为10
	//对数组赋值
	heap[1] = 85, heap[2] = 55, heap[3] = 82, heap[4] = 57, heap[5] = 68;
	heap[6] = 92, heap[7] = 99, heap[8] = 98, heap[9] = 66, heap[10] = 56;
	heapSort();//堆排序
	//打印输出堆排序后的数组
	for (int i = 1; i <= n; i++) {
		cout << heap[i] << " ";
	}
	cout << endl;//换行

	return 0;
}