算法思想:建立一个最大堆(最小堆),最大堆满足一个性质,即所有的父结点都要比它的左右结点大,故可以每次将根结点抽取出来放在最后面,每次都抽取剩余部分的最大值,那最终得到的序列就是有序的
建立最大堆的方法:从最底层的父结点开始,从下往上,不断的把子树调整成最大堆,最后根结点就是最大元素,然后跟最后面的元素交换位置,交换完再重新进行调整
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int N = 1e5;
int a[N];
void HeapAdjust(int i, int size)
{
int left = 2 * i; //左孩子节点
int right = 2 * i + 1; //右孩子节点
int mx = i; //暂存最大元素下标
if(left <= size && a[left] > a[mx])
mx = left;
if(right <= size && a[right] > a[mx])
mx = right;
if(mx != i) //当前父结点不是最大的
{
swap(a[i], a[mx]);
HeapAdjust(mx, size); //递归处理子树
}
}
void BuildHeap(int size)
{
for(int i = size / 2; i >= 1; i--) //非叶结点最大下标为size/2
HeapAdjust(i, size);
}
void HeapSort(int size)
{
BuildHeap(size); //建立初始最大堆
for(int i = size; i >= 2; i--)
{
swap(a[1], a[i]); //把最大元素放在最后面
size--; //不包括最大元素
HeapAdjust(1, size); //重新调整
}
}
int main()
{
int n;
cin >> n;
for(int i = 1; i <= n; i++)
scanf("%d", &a[i]);
HeapSort(n);
for(int i = 1; i <= n; i++)
printf("%d ", a[i]);
return 0;
}