今天跟着小编一起学习堆排序吧!
目录
一.实现原理
堆排序的实现模型是一棵二叉树。
(1)我们以升序排序为例,称为大堆排序,把数据随机放入二叉树中,从树尾(最后一棵子树)开始排序,左右子树大的与双亲节点比较,谁大谁成为双亲节点。
(2)把这左右子树同一层的按(1)步骤排序,都排完后进入上一层尾开始排。
(3)所有的层都排完后,树头(根节点)就是这颗二叉树中的最大数,将它与树尾交换。注意:树尾其实就是数组的末端,就是把最大值放入数组末端。
(4)我们再从树尾开始,不过是从树尾的前一个节点开始,因为树尾已经是最大值了。以此类推,直到一趟排完后,该从树头开始为止,排序就结束了。
当然,如果是降序排序,就是小堆排序,就选出最小值放入树头,再与树尾交换就行。
堆排序的时间复杂度是O(n*logn),所以效率相对而言比较高。
当然,小编也为大家准备了具体的流程图,相信看完图大家就能完全理解了!
上图:
二.代码实现
//此处用升序举例,使用大堆排序,若为降序则是小堆
void hampsort(int* a, int n)//主函数
{
for (int tail = (n - 1 - 1) / 2; tail >= 0; tail--)//从最后一颗子树开始进行排序
{
hampsort_base(a, n, tail);
}
int end = n - 1;
while (end)
{
swap(&a[end], &a[0]);
hampsort_base(a, end, 0);//再进行子树的大堆排序,确保子树的根节点是子树的最大值
end--;
}
print(a, n);
}
void hampsort_base(int* a, int n, int root)//子函数,n是数组的长度,root是根节点
{
int par = root;//父节点所在的数组下标
int child = root * 2 + 1;//leftchild,左孩子节点所在的数组下标
while (child < n)//孩子节点下标最多为n-1
{
// child<n-1是因为之后有需要使用child+1,若child为n-1则存在越界
if (child < n - 1 && a[child] < a[child + 1])//先进行左右孩子节点的比较
{
child += 1;
}
if (a[child] > a[par])//孩子节点较大值与父节点比较,若大于则交换
{
swap(&a[child], &a[par]);
par = child;//子树的子树进行大堆排序
child = par * 2 + 1;
}
else
{
break;
}
}
}
void swap(int* a, int* b)//交换数值
{
int x = *a;
*a = *b;
*b = x;
}
创作不易,感谢三连!如有错误,敬请斧正😀