前言
最近在复习的时候,复习到了堆排序,简单了解后,做一个代码总结。
注:本文章只对功能做一个实现。参考文献贴在文章末尾。
完整代码
#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;
}