堆排序分为两个阶段:1、将原始数组组装成一个堆;2、从堆顶逐个取出元素并得到排序结果。(如果是最大堆,则是重复删除最大元素,然后从后往前放入到数组。)
用sink()只需扫描数组中的一半元素。
堆排序可以不需要额外的空间,最优的利用空间和时间。可用于嵌入式系统。缺点:无法利用缓存,影响缓存命中。
#include <iostream>
using namespace std;
int parent(int i)
{
return i/2;
}
int left(int i)
{
return 2*i;
}
int right(int i)
{
return 2*i+1;
}
void swap(int a[],int i,int j)
{
int temp=a[i];
a[i]=a[j];
a[j]=temp;
}
void swim(int a[],int k) {}// 用swim的话,则在heap_sort中从左至右遍历数组,并swim
void sink(int a[],int k,int n)// 用sink只需扫描数组中的一半元素
{
while(2*k<=n)
{
int j=left(k);
if( right(k)<=n && a[right(k)]>a[left(k)]) j=right(k);
if( a[k] > a[j]) break;
swap(a,k,j);
k=j;
}
}
void heap_sort(int a[],int n){
// 将数组构造为堆
for(int k=n/2;k>=1;k--){
sink(a,k,n);
}
// 在获取剩余最大元素的过程中销毁堆
while(n>1){
swap(a,1,n--);
sink(a,1,n);
}
}
void show(int a[],int n){
for(int i=1;i<=n;i++){
cout<<a[i]<<" ";
}
cout<<endl;
}
int main(){
int a[11]= {-1,3,6,2,7,9,0,8,1,4,5};
// 第0个元素不计入排序
// 若考虑需相应修改left right parent 及数组遍历的边界即可
show(a,10);
heap_sort(a,10);
show(a,10);
return 0;
}
/* Test example
3 6 2 7 9 0 8 1 4 5
0 1 2 3 4 5 6 7 8 9
Process returned 0 (0x0) execution time : 0.346 s
Press any key to continue.
*/