实现堆排序的代码如下:
#include<stdio.h>
void swap(int arr[],int i,int j)//做交换的一个函数
{
int temp=arr[i];
arr[i]=arr[j];
arr[j]=temp;
}
void heapify(int tree[],int n,int i)//比较,交换 ,保证次节点往下是一个堆
{
//递归出口
if(i>n){
return ;
}
int c1 = 2*i+1;
int c2 = 2*i+2;
int max=i;
if(c1<n && tree[c1]>tree[max]) {//必须保证c1,c2不会出界
max=c1;
}
if(c2<n && tree[c2]>tree[max]) {
max=c2;
}
if(max!=i){
swap(tree,max,i); //做交换
heapify(tree,n,max);//递归
}
}
void build_heap(int tree[],int n){//对乱序的完全二叉树构建堆
int last_node=n-1;
int parent=(last_node-1)/2;
int i;
for(i=parent;i>=0;i--)//从第一个非叶节点开始,建队heapify
{
heapify(tree,n,i);//
}
}
void heap_sort(int tree[],int n)//堆排序 (有了一个堆之后,如何进行堆排序)
{
build_heap(tree,n);//建立一个堆
int i;
for(i=n-1;i>=0;i--){
swap(tree,i,0);//把当前的第一个节点和最后一个节点做交换
heapify(tree,i,0);// 重新调整为一个完整的堆
}
}
int main()
{
int tree[]={2,5,3,1,10,4};
int n=6;
heap_sort(tree,n);//建立一个大顶堆(函数里面使用了递归)
int i;
for(i=0;i<n;i++)
{
printf("%d\n",tree[i]);
}
return 0;
}
//那么如何使用大顶堆进行排序呢?
/*
首先需要将第一个节点和最后一个节点 做一次交换,砍断最后一个节点,对第一个节点做heapify,
重复执行如上操作,直到最后大顶堆只剩下一个节点。
*/