堆排序算法
开始时,堆排序算法先用BuildMaxHeap 将输入数组data[0...n-1]构造成一个最大堆。因为数组中最大元素在根data[0],则可以通过把它与data[n-1]互换来达到最终正确的位置。但是新的根元素可能违背了最大堆性质。这时调用MaxHeapify(data,0,size)就可以保持这一性质,在data[0...n-2]中构造出最大堆。对排序算法不断重复这个过程,堆的大小由n-1 一直降到 2,排序完毕。
//Heap Sort
//by yew1eb
#include <iostream>
#include <algorithm>
using namespace std;
#define N 1000
/*
*堆化
*MaxHeapify让a[i]在堆中"下降",使以i为根的子树保持堆的性质。
*/
void MaxHeapify(int *data, int i, const int size) {
int p = i*2 + 1;
while ( p<size ) {
if( p+1<size ) {
if( data[p]>data[p+1] )
++p;
}
if(data[i]>data[p] ) {
swap(data[p], data[i]);
i = p;
p =i*2 + 1;
} else
break;
}
}
void BuildMaxHeap(int *data, int size) {
for(int i=size/2; i>=0; --i)
MaxHeapify(data, i, size);
}
void HeapSort(int *data, int size) {
BuildMaxHeap(data,size);
for(int i = size -1; i>0; --i) {
swap(data[0], data[i]);
MaxHeapify(data,0,i);
}
}
void PrintArray(int *data, int size) {
for(int i=0; i<size; ++i)
cout<<data[i]<<" ";
cout<<endl;
}
int main() {
int size;
int data[N];
cin>>size;
for(int i=0; i<size; ++i)
cin>>data[i];
HeapSort(data,size);
PrintArray(data,size);
return 0;
}
练习题: HDU 1040 As Easy As A+B 我的代码