当初学习堆排序的时候以为是多大的洪水猛兽,但实质上去学习的时候,其实堆排序的原理很简单,就是一个堆建好后,拿去根节点(因为大顶堆/小顶堆中根节点最大/小),然后剩下的元素再建成一个堆,重复上面的过程,不断取出根节点剩下的节点再建堆,直至最后当所有的节点都被取出之后就成有序的了。
/*********** 堆排序 ***********/
void Swap(double &arr1, double &arr2)
{
double temp = arr1;
arr1 = arr2;
arr2 = temp;
}
// 调整堆
int HeapAdjust(double *arr, int dad,int m)
{
for (int son = 2 * dad + 1; son < m; son = 2 * dad + 1)
{
if (son + 1 < m && arr[son] < arr[son + 1])
++son;
if (arr[dad] > arr[son])
return 0;
Swap(arr[dad], arr[son]);
dad = son;
}
return 0;
}
// 堆排序
int HeapSort(double *arr, int arrLenth)
{
for (int i = arrLenth/2-1 ; i >=0 ; i--)
HeapAdjust(arr, i, arrLenth); // 建堆
for (int i = arrLenth-1 ; i >=1 ; i--)
{
Swap(arr[0], arr[i]);
HeapAdjust(arr, 0, i - 1);
}
return 0;
}
注意点:千万注意for循环的边界条件,不要数组越界。