Heap Sort ——重拾堆排序

// heap sort.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <algorithm>
using namespace std;

void AdjustMaxHeap(int* array,int pos,int end){//array中除了pos之外均满足堆的定义
//本函数进行调整，使array[pos~n]成为一个大顶堆
int temp = array[pos];
for(int i = 2*pos+1;i<=end;i=i*2+1){
if(i<end && array[i]<array[i+1])
++i;//i记录较大的下标
if(temp>array[i])
break;
array[pos]=array[i];
pos = i;
}
array[pos] = temp;//插入最开始不和谐的元素
}

void HeapSort(int* array,int n){
for(int i = n/2-1;i>=0; i--){ //从第一个非叶结点进行调整
AdjustMaxHeap(array,i,n-1);
}
for(int i = n-1;i>=0;i--){
swap(array[0],array[i]);
AdjustMaxHeap(array,0,i-1);
}
}

int _tmain(int argc, _TCHAR* argv[])
{
int a[]={5,3,2,1,4,9,2,6};
HeapSort(a,8);
return 0;
}