上图转自百度百科。
#include <iostream>
using namespace std;
void max_heap(int *data,int i,int size)
{
int max=i;
int temp;
if((2*i+1)<size && data[i]<data[2*i+1])
{
max=2*i+1;
}
if((2*i+2)<size && data[max]<data[2*i+2])
{
max=2*i+2;
}
if(max!=i)
{
temp=data[i];
data[i]=data[max];
data[max]=temp;
max_heap(data,max,size);
}
}
void build_max_heap(int *data,int size)
{
int start;
if(size%2==0)
start=size/2-1;
else
start=(size-1)/2-1;
while(start>-1)
{
max_heap(data,start,size);
--start;
}
}
void heap_sort(int *data,int size)
{
build_max_heap(data,size);
int temp;
for(int i=0; i<size-1; ++i)
{
temp=data[0];
data[0]=data[size-1-i];
data[size-1-i]=temp;
max_heap(data,0,size-i-1);
}
}
int main()
{
int data[5]={3,2,7,5,9};
heap_sort(data,5);
for(int i=0;i<5;++i)
{
cout<<data[i]<<endl;
}
return 0;
}