思想:利用二叉大根堆堆顶元素为最大值,把最大值取出后的元素再构建二叉大根堆,取次大值的方法,从而得到有序排列!
下面代码可直接运行:
#include <stdio.h>
#include <stdlib.h>
inline void swap(int & a, int & b)
{
int c=a;
a=b;
b=c;
}
void print_array(int array[], int num)
{
for(int i=0; i<num; i++)
printf("%d ", array[i]);
printf("\n");
}
//a[0]...a[num-1]
void siftdown(int array[], int i, int num)
{
int temp=array[i];
int child=i;
while(2*i+1<num)
{
child=2*i+1;
//保证child是最大的子节点
if(child+1<num && array[child+1]>array[child])
child=child+1;
//如果子节点都比该节点值小,说明不用再下筛
if(array[child]>temp)
array[i]=array[child];
else
break;
i=child;
}
//array[child]=temp;
array[i]=temp;
}
//a[0]...a[num-1]
void heap_create(int array[], int num)
{
int nIter=num/2;
for(; nIter>=0; nIter--)
siftdown(array, nIter, num);
}
void heap_sort(int * array, int num)
{
heap_create(array, num);
printf("创建好的堆如下:\n");
print_array(array, num);
while(num>=2)
{
swap(array[num-1], array[0]);
printf("第%d次之后的堆如下:\n", 9-num);
print_array(array, num);
num--;
siftdown(array, 0, num);
}
}
void main()
{
int array[]={0,30,155,1,80,300,170,40,99};
print_array(array, 9);
heap_sort(array, 9);
print_array(array, 9);
}
堆排序的步骤:
1. 把已给的数组变成一个堆。时间复杂度为:n
使用方法:对n/2 to 0依次进行下筛,每次下筛时间复杂度为:lgn
2. 把堆顶的元素(数组首位)与数组尾的元素进行交换,即数组尾为最大值。对前面的n-1个元素进行排序即可。
技巧性的方法:利用已经建好的堆。堆顶最大值与数组尾交换,与删除堆顶元素的处理相同,只需对新的堆顶元素下筛即可组成新的堆!
3. 重复步骤2,直到全部都排好!
总得时间复杂度:n+n*lgn=nlogn
空间复杂度:1
时间复杂度与快速排序法相同,但据说快速排序法的实际效果更好些!