C/C++实现堆排序

介绍

堆排序是指利用堆这种数据结构所设计的一种排序算法。
首先，堆不是容器，而是组织容器元素的一种特别方式。堆的逻辑结构与完全二叉树类似。

堆又分为大顶堆和小顶堆
大顶堆： 父节点总是大于或等于子节点
小顶堆： 父节点总是小于或等于子节点

下文利用大顶堆进行堆排序。
 

代码

堆排序，利用大顶堆对数组进行排序，那么我们就得先构造大顶堆。
思路： 把所有非终端结点都检查一遍，也就是把所有父结点遍历一遍，看其是否满足大顶堆要求。

按照上图，从最后一个父结点开始，是下标 i=(数组长度)/2 的元素，也就是元素9；然后依次检查元素78、17、53。（数组实际长度为8，第0位空出）

检查时，从左右孩子中选出较大的替换即可。 下图为元素9调整后的结果：

下图是78调整后的结果：

下图为元素17调整结果：

元素53就要调整两次，53和87互换后，53所在子树仍不满足大顶堆特性，还得再和78互换。

代码如下：

void HeadAdjust(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++;
		if(A[0]>=A[i]) break;
		else{
			A[k]=A[i];
			k=i;
		}
	}
	A[k]=A[0];
}

void BuildMaxHeap(int A[],int len){
	for(int i=len/2;i>0;i--){
		HeadAdjust(A,i,len);
	}
}

上述代码也是堆排序的核心代码。
当把数组构造为大顶堆后，我们选取堆顶元素（即最大元素）与最后元素互换。去除取出的最大元素，将剩余的n-1个元素继续构造为大顶堆，然后再选取最大元素，以此类推。这样就可以得到降序的数组。

选出最大元素（即堆顶元素）与堆中最后元素交换：

除堆中最后一个元素外，剩余元素继续调整为大顶堆，然后重复操作：

void HeapSort(int A[],int len){
	BuildMaxHeap(A,len);
	for(int i=len;i>1;i--){
		cout<<A[1]<<" ";
		swap(A[i],A[1]);//最后元素A[i]与最大元素A[1]交换
		HeadAdjust(A,1,i-1);//剩余元素继续调整为大顶堆
	}
	cout<<A[1]<<" ";
}

 
完整代码如下：

#include <iostream>

using namespace std;

void HeadAdjust(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++;
		if(A[0]>=A[i]) break;
		else{
			A[k]=A[i];
			k=i;
		}
	}
	A[k]=A[0];
}

void BuildMaxHeap(int A[],int len){
	for(int i=len/2;i>0;i--){
		HeadAdjust(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]);
		HeadAdjust(A,1,i-1);
	}
	cout<<A[1]<<" ";
}

int main(){
	int n;
	cin>>n;
	int a[n+1];
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	HeapSort(a,8);
	return 0;
}