堆排序结合了归并排序O(nlogn)时间复杂度的优点以及插入排序原地排序的优点,算法原理也很清晰:通过max_heapify不断比较,保持根结点最大来建立一个大顶堆。建堆完成后,把根结点与最后一个位置交换,同时整个堆的线性存储长度减一,再利用max_heapify保持“根结点最大”的性质,如此反复。最后一次交换前,堆中只剩下两个结点,交换后,排序完成,顺序是从小到大。代码如下:
#include <stdio.h>
#include <conio.h>
//函数原型
void max_heapify(int A[],int i,int length);
void build_max_heap(int A[]);
void heap_sort(int A[]);
void swap(int *a,int *b);
//定义全局变量
int n=10;//需要排序的元素的个数
int main()
{
int test[11]={-1,5,4,6,9,8,7,2,3,1,10};//除test[0]外,10个测试数据
//排序
heap_sort(test);
//输出
for(int i=1;i<=10;i++)
printf("%d ",test[i]);
getch();
}
//保持大顶堆性质
void max_heapify(int A[],int i,int length)
{
int largest=i; //最大值的下标
int sub_left=i<<1;//左子树根结点的坐标
int sub_right=sub_left+1;//右子树根结点的坐标
if(sub_left<=length && A[i]<A[sub_left])
largest=sub_left;
if(sub_right<=length && A[largest]<A[sub_right])
largest=sub_right;
if(largest!=i)
{
swap(&A[largest],&A[i]);
max_heapify(A,largest,length);
}
}
//建堆
void build_max_heap(int A[])
{
for(int i=(n>>1);i>=1;i--)
max_heapify(A,i,n);
}
//堆排序
void heap_sort(int A[])
{
int length=n;
build_max_heap(A);
for(int i=n;i>=2;i--)
{
swap(&A[i],&A[1]);
length--;
max_heapify(A,1,length);
}
}
//交换
void swap(int *a,int *b)
{
int temp=*a;
*a=*b;
*b=temp;
}