#include <stdio.h>
#include <stdlib.h>
/*由于不会动态获得当前堆的元素数量heap_size,所以暂时用传参的方法,但是当多次运行后发现
为了防止错误应该把heap_size设置为全局变量,主函数初始化后,让它被各个函数操作过程中动态
改变也是可以的。
*/
int left(int i)//返回左孩子位置
{
return 2*i;
}
int right(int i)//返回右孩子位置
{
return 2*i+1;
}
void max_heapify(int *a,int heap_size,int i)//保持堆性质,使以i为根的子树成为最大堆 ,heap_size当前堆中元素数量
{
//假设左右孩子已经分别为最大堆
//l,r分别为左右孩子的位置,largest用于临时存放当前节点与左右孩子中最大的节点的位置
int l,r,largest;
int temp;
l=left(i);
r=right(i);
if(l<=heap_size&&a[l]>a[i])//当做孩子大于父节点时,左孩子设为最大,否则父节点最大
largest=l;
else
largest=i;
if(r<=heap_size&&a[r]>a[largest])//继续判断,当右孩子大于最大时,右孩子设为最大,否则刚才的最大不变
largest=r;
if(largest!=i)//如果largest变化了,则通过交换使当前最大节点为父节点
{
temp=a[i];
a[i]=a[largest];
a[largest]=temp;
max_heapify(a,heap_size,largest);//继续递归保持交换后的某一孩子的堆的性质。
}
}
void build_max_heap(int *a,int length)//建立最大堆
{
/*从第一个叶子的前一个节点开始循环维护最大堆,由于这个节点后的都是叶子,单一元素可以当做最大堆。
这个叶子往前循环,保证每个处理后的节点的子节点都已经是最大堆,一直到根节点。*/
int i,heap_size;
for(i=length/2;i>=1;i--)
max_heapify(a,length,i);
}
void heap_sort(int *a,int length)//堆排序,从小到大 ,length为堆中元素数量
{
int heap_size=length;
int i,temp;
build_max_heap(a,length);//建立最大堆
for(i=length;i>=2;i--)//每次把最大堆顶部的元素,即当前的最大值放入数组末尾,然后维护最大堆性质。
{
temp=a[i];
a[i]=a[1];
a[1]=temp;
heap_size--;
max_heapify(a,heap_size,1);//重新调整为最大堆
}
}
int main(int argc, char *argv[])
{
// int a[11]={0,4,1,3,2,16,9,10,14,8,7};//测试数据
//int length=10;
int length;//数组中的堆的数据的长度
int a[100];
a[0]=0;
int i;
printf("请输入元素的数量:");
scanf("%d",&length);
printf("\n请依次输入%d个元素,空格结束:\n",length);
for(i=1;i<=length;i++)
scanf("%d",&a[i]);
printf("\n");
for(i=1;i<=length;i++)
printf("%d ",a[i]);
printf("\n建立最大堆以后\n");
build_max_heap(a,length);
for(i=1;i<=length;i++)
printf("%d ",a[i]);
printf("\n");
printf("使用堆排序后:\n");
heap_sort(a,length);
for(i=1;i<=length;i++)
printf("%d ",a[i]);
system("PAUSE");
return 0;
}
算法入门--堆排序(最大堆,从小到大排序)
最新推荐文章于 2024-04-28 10:27:05 发布
