目录
本文章所用图(除代码截图外)来源与:
作者: dreamcatcher-cx
出处: <http://www.cnblogs.com/chengxiao/>
1、堆的概念:
首先堆必须是一个完全二叉树,堆又分为大顶堆和小顶堆。大顶堆:每个节点的值大于或等于其左右孩子节点的值;小顶堆:每个节点的值都小于或等于其左右孩子的值。
虽然我们称这种结构为堆,但只是我们在空间想象的这种结构,在实际操作或代码中,我们还是以数组的方式进行操作,即根节点为第一个元素,它的左孩子为第二个元素,右孩子为第三个元素,依次遍历每个节点。如下图
2、堆排序算法
首先我们要将一个无序的数组调整成一个有序的大顶堆,这里我们要记住一个关系式
下标为i的元素的父节点为:(i-1)/ 2;
下标为i的元素的左孩子为:2*i+1;
下标为i的元素的右孩子为:2*i+1;
在整理成堆的时候,我们要从下到上、从右到左的去遍历每个节点的左孩子和右孩子,如果左孩子或右孩子比父节点大,我们就让大的那个元素下标为父节点,让循环往下走就是了,只要有左孩子或者右孩子比父节点大,我们就进行以上的操作。我们来看代码
void heapadjust(int *__src,int length,int i)//参数为(原数组,数组长度,待整理的节点下标)
{
int max = i;
int lchild = i*2+1;
int rchild = i*2+2;
if(lchild < length && __src[max] < __src[lchild])
max = lchild;
if (rchild < length && __src[max] < __src[rchild])
max = rchild;
if(max != i)
{
swap(&__src[max],&__src[i]);
heapadjust(__src,length,max);
}
}
完成了大顶堆的整理过后,我们再来进行堆排序。
堆排序就是我们整理好后的大顶堆,它的根节点一定是最大的,我们让根节点与末尾元素进行交换,交换过后让最后一个元素也是最大的元素移动,元素个数减一,再进行递归。这样就完成了堆排序。
void heapsort(int *__src,int length)//参数为(原数组,数组长度)
{
int i;
for(i=length/2-1;i>=0;i--)//第一个for循环将原数组整理成大顶堆的结构
heapadjust(__src,length,i);
for(i=length-1;i>0;i--)//第二个for循环将将根节点与最后一个元素交换,并使堆结构的元素自减
{
swap(&__src[i],&__src[0]);
heapadjust(__src,i,0);
}
}
其他函数
void produce_arr(int *__src,int length)//用随机数对数组进行赋值
{
srand(time(NULL));
for(int i = 0;i<length;i++)
__src[i] = rand()%100;
}
void traversal_arr(int *__src,int length)//对数组元素进行输出
{
for (int i = 0; i < length; i++)
{
printf("%d ",__src[i]);
}
printf("\n");
}
void swap(int *__src,int *__dest)//交换两个数的值
{
int temp = *__src;
*__src = *__dest;
*__dest = temp;
}
主函数及运行结果
#include"sort.h"
#define LENGTH 10
int main()
{
int arr[LENGTH];
produce_arr(arr,LENGTH);
printf("the original array is:\n");
traversal_arr(arr,LENGTH);
heapsort(arr,LENGTH);
printf("after the heapsort is:\n");
traversal_arr(arr,LENGTH);
return 0;
}