1.什么是堆?
堆其实是一颗完全二叉树,堆分为大堆和小堆。所谓大堆,就是父节点的值大于等于左右孩子结点的值,即a[i] >= a[2*i+1] && a[i] >= a[2*i+2]。所谓小堆,就是父节点的值小于等于左右孩子结点的值,即a[i] <= a[2*i+1] && a[i] <= a[2*i+2]。
堆排序是用堆这种数据结构设计的一种排序算法,是选择排序的一种。利用数组的特点快速定位到指定索引的元素。
2.算法思想
首先建一个大堆,堆顶元素是最大的一个元素。end指向堆的最后一个节点,堆顶元素和end指向的元素交换,则最大的元素到了最后一个位置。end--,堆顶元素进行一次向下调整,调整完毕,堆顶元素为次大元素,和end指向的元素交换。直至end=0结束。此时,堆中的元素按照下标值是有序的。
void AdjustDown(int* a,int root,size_t n)
{
size_t parent = root;
size_t child = parent*2+1;
while(child < n)
{
if(child+1 < n && a[child+1] > a[child])
++child;
if(a[child] > a[parent])
{
swap(a[child],a[parent]);
parent = child;
child = parent*2+1;
}
else
{
break;
}
}
}
void HeapSort(int* a,size_t n)
{
//建大堆
for(int i = (n-2)/2; i >= 0; --i)
{
AdjustDown(a,i,n);
}
int end = n-1;
while(end > 0)
{
swap(a[0],a[end]);
AdjustDown(a,0,end);
--end;
}
}
void TestHeapSort()
{
int a[] = {2,9,12,15,20,8,3,29,30};
HeapSort(a,sizeof(a)/sizeof(a[0]));
PrintArray(a,sizeof(a)/sizeof(a[0]));
}
运行结果: