例子: 49,38,25,34,76,15,100 排序 。
其中堆排序的过程:(从小到大)
#include<iostream>
using namespace std;
//调整节点 大根堆
template<class T>
void AdjustHeapNode(T a[],int i,int n){ //调整节点i,数组共有N个节点
if (n==1||i>(n-2)/2) //i为叶子节点 (n-2)/2 最后一个非叶子节点的位置
return;
int iLeft=2*i+1;
int iRight=2*i+2;
if (iRight<=n-1) //说明i有左右两个子节点 三个节点找最大值
{
if (a[i]>=a[iLeft]&&a[i]>=a[iRight]) // i 最大 不用调整
return;
if (a[i]<a[iLeft]&&a[iRight]<=a[iLeft]) // iLeft 最大
{
T temp=a[iLeft];
a[iLeft]=a[i];
a[i]=temp;
AdjustHeapNode(a,iLeft,n);
return;
}
if (a[i]<a[iRight]&&a[iLeft]<=a[iRight]) // iRight 最大
{
T temp=a[iRight];
a[iRight]=a[i];
a[i]=temp;
AdjustHeapNode(a,iRight,n);
return;
}
}else{ // 说明i只有左节点 二个节点找最大值
//iLeft为最后一个节点
if (a[i]>=a[iLeft])
return;
else
{
T temp=a[iLeft];
a[iLeft]=a[i];
a[i]=temp;
AdjustHeapNode(a,iLeft,n);
return;
}
}
}
//建立堆
template<class T>
void CreateHeap(T a[],int n)
{
int iFirst=(n-1)/2; //第一个非叶子节点
for (;iFirst>=0;iFirst--)
{
AdjustHeapNode(a,iFirst,n);
}
}
//堆排序
template<class T>
void HeapSort(T a[],int n)
{
CreateHeap(a,n);
T temp;
for (int i=0;i<n-1;i++)
{
temp=a[n-1-i];
a[n-1-i]=a[0];
a[0]=temp;
AdjustHeapNode(a,0,n-1-i);
}
}
int main()
{
int a[]={49,38,25,34,76,15,100};
int size=sizeof(a)/sizeof(a[0]);
HeapSort(a,size);
return 0;
}
其中这是 从小到大的排序,其实我们已经建立了大顶堆,我们可以每一次删除堆顶元素然后调整该堆,其实就是一个从大到小的排序过程。代码如下:
{
int i=0;
while(i<n)
{
cout<<a[0]<<" ";
a[0]=a[n-1-i];
AdjustHeapNode(a,0,n-1-i);
i++;
}
}