注意,C++语言的数组下标是从0开始,则r[i]的左子树是r[2i+1],右子树是r[2i+2];
#include<stdio.h>
#include <iostream>
#include<iomanip>
using namespace std;
void SiftHeap(int r[],int k,int n){
int i,j,temp;
i=k;
j=2*i+1; //j为i的左子树
while(j<n){
if(j<n-1&&r[j]<r[j+1])
j++;
if(r[i]>r[j])
break;
else{
temp=r[i];r[i]=r[j];r[j]=temp;
i=j;j=2*i+1;
}
}
}
void HeadSort(int r[],int n){
int i,temp;
for(i=(n-1)/2;i>=0;i--)
SiftHeap(r,i,n); //初建堆,最后一个分支的下标是(n-1)/2
for(i=1;i<=n-1;i++){ //重建堆
temp=r[0];r[0]=r[n-i];r[n-i]=temp;
SiftHeap(r,0,n-i);
}
}
void main(){
int r[]={50,23,51,46,85,75,42,35,66,69,452,6};
HeadSort(r,12);
for(int i=0;i<12;i++)
cout<<r[i]<<setw(4);
cout<<endl;
}
运行结果:
时间复杂度:O(n*log2 n)