[算法导论] 6.4堆排序

堆数据结构是一种数组对象 ，如图所示，可以看成是一种完全二叉树，除最后一层外，每一层都是填满的

最大堆是满足下面条件的堆：除了根结点以外的每个结点i,有：A[parent(i)]>=A[i],即任意节点的值都小于其父节点

 

利用最大堆排序时，首先将待排序数组整理成最大堆，然后A[1]与A[n]交换，此时A[n]即为最大值，再整理数组A[1....n-1]，使其成为最大堆，A[1]与A[n-1]交换，重复这个过程。

其中用到的关键一步保持最大堆的性质，下面算法中假定以left(i)和right(i)为根的两棵二叉树都已是最大堆，但这时A[i]可能 小于其子女，保持 最大堆的过程如下：

MAX-HEAPIFY(A, i) 
  l ← LEFT(i) 
  r ← RIGHT(i) 
  if l ≤ heap-size[A] and A[l] > A[i] 
     then largest ← l 
     else largest ← i 
  if r ≤ heap-size[A] and A[r] > A[largest] 
     then largest ← r 
  if largest ≠ i 
     then exchange A[i] ↔ A[largest] 
         MAX-HEAPIFY(A, largest)

堆排序代码：

#include <iostream>
using namespace std;

void constructHeap(int * a,int n);  建最大堆（即根结点是最大值）
void maxHeap(int * a,int i,int n);    ///保持最大堆，堆大小是n,要调整的是第i个节点，其左右子树均已是最大堆

int main()
{
	int a[12]={23,1844,546,2,345,234,546,567,678,345,367,9};
	int n=12;
	constructHeap(a,n);
	for(int i=n-1;i>=1;i--)
	{
		int temp=a[0];
		a[0]=a[i];a[i]=temp;
		maxHeap(a,0,i);
	}

	for(int i=0;i<n;i++)
	{
		cout<<a[i]<<"  ";
	}

	return 0;
}

void constructHeap(int * a,int n)
{
	for(int i=n/2-1;i>=0;i--)
	{
		maxHeap( a,i,n);
	}
}

void maxHeap(int * a,int i,int n)
{
	while(i<=n/2-1)
	{
		int leftChild=i*2+1,rightChild=(i+1)*2,largest=a[i],largestIndex=i;
		if(leftChild<n)
		{
			 largest=a[leftChild],largestIndex=leftChild;
		}

		if(rightChild<n && a[rightChild]>largest)
		{
			largest=a[rightChild];largestIndex=rightChild;
		}
		if(a[i]<largest)
		{
			a[largestIndex]=a[i];
			a[i]=largest;
		}
		else
			return;
		i=largestIndex;
	}

}