算法导论(Introduction to Algorithms)之堆排序(C语言实现)

源码下载地址：

http://download.csdn.net/detail/mr_dodo/4962237

堆排序其重要思想，是保证数组最大堆或最小堆的性质(除根节点以外的每个结点i，均有 A[PARENT(i)] >= A[i] 或 A[PARENT(i)] <= A[i] )。最大堆的最大元素在根部，而最小堆的最小元素在根部。

将数组构造成最大堆或最小堆之后，就可以进行排序了。

以下代码演示 最大堆按升序排列。

其最重要一步（构造最大堆）伪代码如下：

/*
* Max-Heapify
* 
* 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)
*/

其C语言实现代码如下：

void MaxHeapify(int * arraySort,int i)
{
	int l = i*2+1;
	int r = i*2+2;
	int largest = 0;

	if (l < heapSize && arraySort[l] > arraySort[i])
	{
		largest = l;
	}
	else largest = i;

	if (r < heapSize && arraySort[r] > arraySort[largest])
	{
		largest = r;
	}

	if (largest != i)
	{
		SWAP_INT(arraySort[i],arraySort[largest])
		MaxHeapify(arraySort,largest);
	}
}

堆排序函数实现代码如下：

void HeapSort(int * arraySort,int arrayLen)
{
	//build max heap
	int i = arrayLen / 2;
	heapSize = arrayLen;

	for ( ; i >= 0 ; i--)
	{
		MaxHeapify(arraySort,i);
	}

	//sort
	for (i = arrayLen - 1 ; i >= 1 ; i--)
	{
		SWAP_INT(arraySort[0],arraySort[i])
		heapSize--;
		MaxHeapify(arraySort,0);
	}
}

测试代码如下：

	int arrarSort[10] = {4,1,3,2,16,9,10,14,8,7};

 	HeapSort(arrarSort,10);
 
	for (int i = 0 ; i < 10 ; i++)
	{
		printf("%d,",arrarSort[i]);
	}

将上诉代码，略微修改：

	if (l < heapSize && arraySort[l] < arraySort[i])
	{
		largest = l;
	}
	else largest = i;


	if (r < heapSize && arraySort[r] < arraySort[largest])
	{
		largest = r;
	}

即为降序排列：