堆排序:利用堆这种数据结构所设计的一种排序算法
每个结点的值都大于其左孩子和右孩子结点的值,称之为大根堆;每个结点的值都小于其左孩子和右孩子结点的值,称之为小根堆。如下图
图片:
二叉树的性质:
1.父结点索引:(i-1)/2 取整
2.左孩子索引:2i+1
3.右孩子索引:2i+2
其中i为数组的索引,
因此大根堆必须满足Key[i]>=Key[2i+1]&&key>=key[2i+2]
小根堆必须满足Key[i]<=key[2i+1]&&Key[i]<=key[2i+2]
原理示意图:
代码如下:
#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 min_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)
{
cout<<"初始数组";
for (int i = 0; i < len; i++)
cout << arr[i] << ' ';
cout<<endl;
//构造初始堆,i从最後一个父节点开始调整
//len个节点的二叉树可以形成 len / 2 - 1个父节点
//将数组调整成一个大根堆
//必须从最后一个父节点调起,才能保证上层的父节点永远大于下层的父节点
//这一步必须先做,这步还有一个作用,确保下面在进行调整的时候不会出现
// 3
// 7 8
// 9 2 5 6
//的情况而选出8(不是最大的数),因为此时已经保证了父节点一定大于子节点
for (int i = len / 2 - 1; i >= 0; i--)
{
cout << "start = " << i<<"; end = "<<len-1<<endl;
max_heapify(arr, i, len - 1);
for (int i = 0; i < len; i++)
cout << arr[i] << ' ';
cout<<endl;
}
cout<<endl;
//先将第一个元素和已经排好的元素前一位做交换,再从新调整(刚调整的元素之前的元素),直到排序完毕
for (int i = len - 1; i > 0; i--)
{
swap(arr[0], arr[i]);
//i-1:排除最后一个已经排出的最大数
max_heapify(arr, 0, i - 1);
for (int i = 0; i < len; i++)
cout << arr[i] << ' ';
cout<<endl;
}
}
int main()
{
int arr[] = {1, 2, 5, 9, 7, 4, 0, 2, 6,12,42,56,77 };
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;
}
参考文章.