堆排序的基本思想是:将待排序序列构造成一个大顶堆,此时,整个序列的最大值就是堆顶的根节点。将其与末尾元素进行交换,此时末尾就为最大值。然后将剩余n-1个元素重新构造成一个堆,这样会得到n个元素的次小值。如此反复执行,便能得到一个有序序列了
#include <iostream>
#include <cstdlib>
using namespace std;
//打印输出函数
void Display(int *a,int size)
{
for(int i=0;i<size;i++)
{
cout<<a[i]<<" ";
}
cout<<endl<<endl;
}
//计算子左节点下标
inline int Left(int i) //inline:内联函数
{
//i++;
return 2*i;
}
//计算子右节点下标
inline int Right(int i)
{
//i++;
return 2*i+1;
}
//维持大顶堆特性
void MaxHeap(int *a,int i,int size)
{
int l,r,heapSize,largest,temp;
l=Left(i);
r=Right(i);
heapSize=size;
if(l<heapSize&&a[l]>a[i])
{
largest=l;
}
else largest=i;
if(r<heapSize&&a[r]>a[largest])
{
largest=r;
}
if(largest!=i)
{
temp=a[i];
a[i]=a[largest];
a[largest]=temp;
MaxHeap(a,largest,heapSize);
}
}
//建立大顶堆
void CreateMaxHeap(int *a,int size)
{
int heapSize=size;
int lengthArray=size;
for(int i=lengthArray-1;i>=0;i--)
{
MaxHeap(a,i,heapSize);
}
}
//堆排序
void HeapSort(int *a,int size)
{
int temp;
int lengthArray=size;
int heapSize=lengthArray;
CreateMaxHeap(a,heapSize);
for(int i=lengthArray-1;i>=1;i--)
{
temp=a[0];
a[0]=a[i];
a[i]=temp;
heapSize-=1;
MaxHeap(a,0,heapSize);
}
}
//计算父节点下标
int Parent(int i)
{
i++;
return i/2-1;
}
int main()
{
int a[6]={4,7,3,2,9,5};
HeapSort(a,6);
Display(a,6);
return 0;
}
为了用户方便,STL在algorithm文件中为用户提供了一个专门用于创建堆的算法make_heap()和一个专门用于堆排序的算法sort_heap()。由于这两个算法的参数均是迭代器类型,故当vector或deque需要进行排序时,用户可以使用它们。