源码下载地址:
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)
*/
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;
}