堆排序的基本思想是:将待排序序列构造成一个大顶堆,此时,整个序列的最大值就是堆顶的根节点。将其与末尾元素进行交换,此时末尾就为最大值。然后将剩余n-1个元素重新构造成一个堆,这样会得到n个元素的次小值。如此反复执行,便能得到一个有序序列了
#include <iostream>
#include <algorithm>
using namespace std;
void heap_adjust(int arr[], int father, int n)
{
int child = father * 2 + 1; //左子节点
int temp = arr[father]; //保存一下最初的父亲节点,后面交换用到。
while (child < n)
{
if (child + 1 < n && arr[child] < arr[child + 1])
child++; //取两个节点中较大者
if (arr[father] >= arr[child])
break;
arr[father] = arr[child]; //交换子节点和父节点
father = child;
child = father * 2 + 1;
arr[father] = temp;
}
}
void build_heap(int arr[], int n)
{
for (int i = (n - 1) / 2; i >= 0; i--)
{ //建堆时从非叶子节点开始调整
heap_adjust(arr, i, n);
}
}
void heap_sort(int arr[], int begin, int end)
{
build_heap(arr + begin, end - begin); //建立一个大顶堆
for (int i = end - 1; i > begin; i--)
{
swap(arr[0], arr[i]); //每次交换根节点和最后一个元素
heap_adjust(arr + begin, 0, i); //将剩余的元素建成另一给大顶堆
}
}
int main()
{
int arr[110];
int n;
cin >> n;
for (int i = 0; i < n; i++)
{
cin >> arr[i];
}
heap_sort(arr, 0, n);
for (int i = 0; i < n; i++)
{
cout << arr[i] << endl;
}
return 0;
}