C++堆排序简单实现

前言

最近在复习的时候，复习到了堆排序，简单了解后，做一个代码总结。

注：本文章只对功能做一个实现。参考文献贴在文章末尾。

完整代码

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int a[N];
int n;

void print(){
	for(int i = 0; i < n; i++) cout << a[i] << ' ';
	cout << '\n';
}

//调整大顶堆 
void adjust(int fa, int len){
	//记录左子节点，右子节点 
	int l = fa * 2 + 1, r = fa * 2 + 2; 
	//最大节点 
	int largest = fa;
	//如果左子节点大于最大节点，就修改最大节点 
	if(l < len && a[l] > a[largest]) {
		largest = l;
	}
	//如果右子节点大于最大节点，就修改最大节点 
	if(r < len && a[r] > a[largest]) {
		largest = r;
	}
	//如果最大节点不是父节点，那么就交换父节点和最大节点 
	if(largest != fa){
		swap(a[fa], a[largest]);
		//继续调整最大节点 
		adjust(largest, len);
	}
}

//建立大顶堆 
void buildHeap(int len){
	//从最后一个非叶子节点开始，进行调整 
	for(int i = len / 2 - 1; i >= 0; i--){
		adjust(i, len);
	}
}

void HeapSort(){
	//首先建立大顶堆 
	buildHeap(n);
	int len = n;
	for(int i = len - 1; i >= 0; i--){
		//此时已经是大顶堆，只需要把堆顶元素和最后一个元素交换就可以了 
		swap(a[0], a[i]);
		//调整堆顶元素，保持大顶堆 
		//因为交换之前是大顶堆
		//也就是所有根节点比左右子节点大
		//那么只需要调整交换之后的根节点就可以了 
		adjust(0, i);
	}
}

int main(){
	cin >> n;
	for(int i = 0; i < n; i++) cin >> a[i];
	HeapSort();
	for(int i = 0; i < n; i++) cout << a[i] << ' ';
	return 0;
} 

参考文献

1. 菜鸟教程：堆排序
2. 谈谈堆排序，大顶堆，小顶堆？
3. 图解排序算法(三)之堆排序