堆排序是由几种操作共同完成的排序,其中包括对某一根节点(广义的根节点,并不是整棵树最顶端的根节点)的堆顺序调整,这是关键操作。
我们先来认识一下什么是堆顺序,分为大根堆和小根堆,大根堆意为整棵完全二叉树所有的根节点都比其子节点key值要大,小根堆意为整棵完全二叉树所有的根节点都要比其子节点key值要小。故从n/2开始进行堆顺序调整,调整时还要保证其调整后子树依旧为堆顺序,子树的子树依旧为堆顺序,所以就要借助循环,一次一次的向下探索是否需要调整当根节点与其其中一子节点调换时。
大根堆顺序调整:
void AdjustDown(int a[],int k,int len) ///对一棵树根进行大根堆处理,其中包括修正已经成为大根堆的子子树
{
a[0]=a[k];
for(int i=2*k;i<=len;i*=2)
{
if(i<len&&a[i]<a[i+1]) i++; ///找key较大的子节点,假如第i 和 i+1为叶节点,找i和i+1这对兄弟节点中最大的和根节点交换才满足性质呀
if(a[0]>a[i]) break;
else { ///动了哪棵树,就对哪棵树继续进行修正
a[k]=a[i];
k=i;
}
}
a[k]=a[0];
}
void BuildMaxHeap(int a[],int len) ///大根堆实现,其中数组是从1开始,充分利用了完全二叉树的性质
{
for(int i=len/2;i>0;i--) AdjustDown(a,i,len);
}
堆排序(从大到小递减排序):
void AdjustDown(int a[],int k,int len) ///对一棵树根进行大根堆处理,其中包括修正已经成为大根堆的子子树
{
a[0]=a[k];
for(int i=2*k;i<=len;i*=2)
{
if(i<len&&a[i]<a[i+1]) i++; ///找key较大的子节点,假如第i 和 i+1为叶节点,找i和i+1这对兄弟节点中最大的和根节点交换才满足性质呀
if(a[0]>a[i]) break;
else { ///动了哪棵树,就对哪棵树继续进行修正
a[k]=a[i];
k=i;
}
}
a[k]=a[0];
}
void BuildMaxHeap(int a[],int len) ///大根堆实现,其中数组是从1开始,充分利用了完全二叉树的性质
{
for(int i=len/2;i>0;i--) AdjustDown(a,i,len);
}
void HeapSort (int a[],int len)
{
BuildMaxHeap(a,len);
for(int i=len;i>1;i--){
cout<<a[1]<<' ';
swap(a[i],a[1]);
AdjustDown(a,1,i-1);
}
cout<<a[1];
}
这个算法实现是应试写法,其中有很多绕和不简洁的写法,并不适合算法竞赛,应试嘛,尽量不写复杂难懂的算法。其中这个写法的数组很有意思,需要注意一下。AdjustDown中的len为实际待排序元素的数量,并不是最初数组开的大小。若数组开10个,则是从下标1开始算起,总共有9位。
#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
void AdjustDown(int a[],int k,int len) ///对一棵树根进行大根堆处理,其中包括修正已经成为大根堆的子子树
{
a[0]=a[k];
for(int i=2*k;i<=len;i*=2)
{
if(i<len&&a[i]<a[i+1]) i++; ///找key较大的子节点,假如第i 和 i+1为叶节点,找i和i+1这对兄弟节点中最大的和根节点交换才满足性质呀
if(a[0]>a[i]) break;
else { ///动了哪棵树,就对哪棵树继续进行修正
a[k]=a[i];
k=i;
}
}
a[k]=a[0];
}
void BuildMaxHeap(int a[],int len) ///大根堆实现,其中数组是从1开始,充分利用了完全二叉树的性质
{
for(int i=len/2;i>0;i--) AdjustDown(a,i,len);
}
void HeapSort (int a[],int len)
{
BuildMaxHeap(a,len);
for(int i=len;i>1;i--){
cout<<a[1]<<' ';
swap(a[i],a[1]);
AdjustDown(a,1,i-1);
}
cout<<a[1];
}
int main()
{
int a[10]={0,1,2,3,4,5,6,7,8,9};
HeapSort(a,9);
return 0;
}