使用语言:c语言,编译器visual studio 2017
一.基本思想
要明白堆排序,首先要明白完全二叉树的概念。对于一个二叉树,假设它的深度为h,如果除第h层外,其余各层结点数都达到最大,且第h层所有结点都集中在左边,则它是完全二叉树。如图:
堆就是个按顺序存储的完全二叉树。
从图中,我们可以发现:
(1)假设一个元素序号为i,则若它的左孩子存在,序号就是2i+1,右孩子序号2i+2。如上图,数字6的序号为1,它的孩子1、2的序号分别为3、4。
(2)最后一个有孩子的结点序号为n/2-1(n/2向下取整),其中n为元素个数;它前面的所有结点都有孩子;比如上图,最后一个有孩子的结点序号为:10/2-1 = 4。
再补充大根堆和小根堆的概念:
(1)如果每个结点的key值都不大于其孩子结点的key值,则它为小根堆;
(2)如果每个结点的key值都不小于其孩子结点的key值,则它为大根堆。
接下来可以开始进入正题了。以从小到大排序为例,堆排序的基本思想是:
(1)把数据按顺序填入堆;
(2)构造大根堆(原因见下文);具体步骤为:
a.从最后一个有孩子的结点开始,向前遍历所有结点;
b.对于一个结点k,若k的值小于孩子的值,交换k和值最大的孩子的位置;交换完成后,一定注意检查k与新孩子是否满足大根堆;
c.遍历完成,就形成了初始大根堆。
(3)第一个元素现在是值最大的。交换第一个元素和最后一个元素的位置。交换完成后,最大的元素就固定在数组末尾。不管最后一个元素,将前面的所有元素再次构造大根堆,重复执行本步骤,直到构造的大根堆只剩一个元素,排序就完成了。
仍然以上图为例子:
![]() | ![]() |
![]() |
二.代码实现
#include<stdio.h>
void AdjustHeap(int array[], int i, int length) {//调整操作
int parent = i;
int child = 2 * parent + 1;
int temp = array[parent];
while (child < length) {
if (child + 1 < length&&array[child] < array[child + 1]) {//右孩子存在并且值大于左孩子
child++;
}
if (temp >= array[child]) //注意是用temp比较
break;//父结点值已经大于孩子结点,跳出循环
array[parent] = array[child];
parent = child;
child = child * 2 + 1;//继续向下检测是否满足大根堆
}
array[parent] = temp;
}
void HeapSort(int array[],int length) {
int i = length / 2 - 1;//最后一个有孩子的结点
for (i; i >= 0; i--) {
AdjustHeap(array, i, length);//建立初始的大根堆
}
for (int j = length - 1; j >= 1; j--) {
int temp = array[0];//交换array[0]和最后一个元素
array[0] = array[j];
array[j] = temp;
AdjustHeap(array, 0, j);//只需进行一步堆调整,就可得到大根堆
printf("第%d趟排序:",length-j);
for (i = 0; i < length; i++) {
printf("%d ", array[i]);
if (i == length - 1)
printf("\n");
}
}
}
int main() {
int array[100];
int i = 0;
int num = 0;
printf("请输入待排序的数字个数:\n");
scanf_s("%d", &num);
printf("请输入%d个数字:\n", num);
for (i; i<num; i++) scanf_s("%d", &array[i]);
HeapSort(array, num);
return 0;
}
运行结果截图:
三.时间复杂度分析:
这里只给出结论:时间复杂度为O(NlogN)。至于过程,大家可移步参考loveliuzz的博客