一、什么是堆?
堆是一种任意节点不大于或者不小于父节点的完全二叉树结构。
由堆的定义可以知道堆的父节点一定大于或者小于其孩子节点,同时每个结点的左子树和右子树都是一个二叉堆(都是最大堆或最小堆)。其中根节点最大的堆叫做最大堆或大根堆,根节点最小的堆叫做最小堆或小根堆。
二、堆排序基本思想
1) 建立最大堆(最小堆),此时堆顶元素是该组数据中最大数据(最小数据)
2) 移除堆顶元素,重新建立最大堆(最小堆)
3) 重复步骤2)
4) 数据全部移除,移除数据构建成有序序列
三、示例分析 (arr[] = {8, 6, 1, 4, 0, 3, 9, 5, 7, 2})
四、代码实现
#include <iostream>
#include <assert.h>
using namespace std;
void AdjustDown(int* arr, int root, int n)
{
int parent = root;
int child = parent * 2 + 1;
while (child < n) { // 选孩子里面大的一个
if (child + 1 < n && arr[child + 1] > arr[child]) {
++child;
}
if (arr[parent] < arr[child]) {
swap(arr[parent], arr[child]);
parent = child;
child = parent * 2 + 1;
} else {
break;
}
}
}
void HeapSort(int* arr,int size)
{
assert(arr);
// 建大堆
for(int i = (size-2)/2; i >= 0; --i) {
AdjustDown(arr, i, size);
}
// 排序
int end = size-1;
while (end > 0) {
swap(arr[0], arr[end]);
AdjustDown(arr, 0, end);
--end;
}
}
int main(int argc,const char* argv[])
{
int arr[] = {8, 6, 1, 4, 0, 3, 9, 5, 7, 2};
HeapSort(arr, sizeof(arr)/sizeof(arr[0]));
for (int i = 0; i < sizeof(arr)/sizeof(arr[0]); i++) {
cout<<arr[i]<<" ";
}
cout<<endl;
return 0;
}