堆的定义: 堆实际上是一棵完全二叉树。
堆满足两个性质:
1、堆的每一个父节点都大于(或小于)其子节点;
2、堆的每个左子树和右子树也是一个堆。
堆的分类:
1、最大堆(大顶堆):堆的每个父节点都大于其孩子节点;
2、最小堆(小顶堆):堆的每个父节点都小于其孩子节点;
堆的存储:
一般都用数组来表示堆,i结点的父结点下标就为(i – 1) / 2。它的左右子结点下标分别为2 * i + 1和2 * i + 2。
如下图所示:
堆排序:
堆的第一个元素要么是最大值(大顶堆),要么是最小值(小顶堆),这样在排序的时候(假设共n个节点),直接将第一个元素和最后一个元素进行交换,然后从第一个元素开始进行向下调整至第n-1个元素。
所以,如果需要升序,就建一个大堆,需要降序,就建一个小堆。 、
堆排序的步骤分为三步:
1、建堆(升序建大堆,降序建小堆);
2、交换数据;
3、向下调整。
假设我们现在要对数组arr[]={8,5,0,3,7,1,2}进行排序(降序):
首先要先建小堆:
堆建好了下来就要开始排序了: (交换数据+调整堆)
代码实现:
#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)
{
int c1=2*i+1;
int c2=2*i+2;
int max=i;
if(c1<n && tree[c1]>tree[max])
{
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(tree,n,i);
}
}
//堆排序
void heap_sort(int tree[],int n)
{
build_heap(tree,n);
int i;
for(int 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;
}
运行结果: