来源:https://www.runoob.com/w3cnote/heap-sort.html
原文的动图解释的很明白。
代码:
#include <iostream>
#include <algorithm>
using namespace std;
void max_heapify(int arr[], int start, int end) {
//建立父节点下标和子节点下标
int dad = start;
int son = dad * 2 + 1;//父节点与其左节点的关系(水平遍历)
while (son <= end) { //若子节点下标在范围内才继续比较
if (son + 1 <= end && arr[son] < arr[son + 1]) //先比较两个子节点大小,选择其中最大的一个
son++;
if (arr[dad] > arr[son]) //如果父节点大于子节点,表示调整完成,跳出函数
return;
else { //否则交换父子节点内容,继续子节点与孙节点的比较和调整
swap(arr[dad], arr[son]);
//*********把较大的升上去之后,掉下来的较小的同样需要不断下降到合适的位置*************//
dad = son;
son = dad * 2 + 1;
}
}
}
void heap_sort(int arr[], int len) {
//初始化,按照二叉树的关系,i表示二叉树最右下角的父节点,也就是最后一个父节点
for (int i = len / 2 - 1; i >= 0; i--)
max_heapify(arr, i, len - 1);//1、从叶子节点开始,将所有的子节点都小于父节点
//将之前调整之后的二叉树的第一个元素(最大或最小:这里是最大)和最后一个元素交换,继续排序
for (int i = len - 1; i > 0; i--) {
swap(arr[0], arr[i]);//2、从最开始的父节点开始,将所有最大值换到最后的节点(最小值节点)
//3、然后原本最大值节点变成最小值节点,通过while(son<=end)之后的不断操作将,当前的最小
//值移位到最后,并将最大值升至原本最大值节点,重复过程2
max_heapify(arr, 0, i - 1);
}
}
int main() {
int arr[] = { 7,3, 5, 3, 0, 8, 6, 1, 5, 8, 6, 2, 4, 9, 4 };
int len = (int)sizeof(arr) / sizeof(*arr);
heap_sort(arr, len);
for (int i = 0; i < len; i++)
cout << arr[i] << ' ';
cout << endl;
return 0;
}
关键: