数据结构与算法堆排序

Posted 2021-04-25 码农有道

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了数据结构与算法堆排序相关的知识，希望对你有一定的参考价值。

月

日

周

二

面试官：写一个堆排吧

我心想：幸好昨天刚看

堆排是基于堆的一种排序算法，如果对堆的插入和删除不清楚，强烈建议先看、，今天我们聊聊堆排的思想，复杂度以及稳定性

堆排思想

前情回顾：克给谦子解决了时间管理上的问题

过了几天后，谦子高兴地跑到老师跟前

老师，你的那个方法太好了,现在做事非常有次序

谦子

克

那就好，看来你已经理解了二叉堆了，今天你刚好来，顺便考考你

早知不来了【数据结构与算法】堆排序，谦子心想

克

给你一个无序数组，你给把它排好序

谦子心想：上次去溪边游玩不是已经出过这个题吗，当时学会了冒泡排序，这次肯定不是那么简单，肯定和前几天的堆有关系

这个应该用到堆

谦子

克

恩恩，对，怎么用堆这个数据结构来排好一个数组中的元素呢？

如果将这个数组调整为小根堆，那么根据小根堆堆顶一直是最小元素的特性，我可以不断地取（删除）堆顶的元素，直到堆中只剩下一个元素，这样就可以得到一个递减的元素序列了

谦子

删除中交换操作会破坏堆有序

但下沉操作会重新调整堆

谦子画了一个图解释道

这样一直删除到只有一个元素的时候，整个数组就排好序了

谦子

删除包括三个步骤：

① 交换堆顶与堆最后一个元素

② 堆大小(heapSize)减一

③ 调整堆(sink(arr,1))

具体讲解请看:

克

你刚才是建立在数组是小根堆的情况下进行的，那如何将无序数组调整成堆呢？

建堆

我可以遍历数组中的每个元素，给一个空堆不断插入元素，直到插完所有元素，这样就可以构造一个堆了

谦子

上浮(swim)操作会保持堆有序

这样一直把数组中的元素插完就建成一个小根堆了

谦子

克

恩恩，这确实是一种方法，但更常见的是用下沉(sink)操作自底向上构造一个堆

哦，这个怎么做？

谦子

克

初始的时候，无序数组逻辑上可以看成一颗完全二叉树

克

每个叶子节点可以视为一个大小为一的堆，我们可以自底向上从非叶子节点开始每层从右至左给每个节点都调用下沉(sink)方法，这样以当前节点为根节点的树就变为堆了

克

然后给左边的节点4执行下沉(sink)操作（同一层从右向左）

克

最后根节点执行下沉(sink)操作

克

这样整个完全二叉树就变成了一个最小堆

注意：在下沉(sink)某个节点的时候，这个节点的两个孩子必须是堆

堆排代码

太神奇了，还可以这样

谦子

克

恩恩，那你写一写建堆和删除最小值的代码吧

克突然话锋一转

哦！好吧

谦子

又要写代码，谦子心想，虽说心中这样想，但还是要写的，思考了一会，只见谦子写下了如下代码

这里的 heapSize/2 就是第一个非叶子节点的下标

谦子

谦子解释道

过了一会谦子又写出了删除最小值的代码

克

恩恩，不错，那顺便写一下排序的代码吧

前两个都写出来了，剩下的排序就很简单了，按照之前的思路，先建立一个小根堆，然后不断地删除堆顶最小元素，删除N-1次就OK了

只需删除N-1次，剩下的那个自然是最大的，所以我循环N-1次

谦子

谦子解释道

克

恩恩，很好，这个排序就是今天要给你说的另一个排序：堆排序

谦子暗自惊叹老师的功力，不知不觉又学到了一个排序方法

时间复杂度

克

那你分析一下这个堆排序的时间复杂度吧

看到数学头疼的可以直接跳过看结论

谦子还没缓过神，又来了一个最让他头疼的时间复杂度

这个。。。，弟子不才，还请老师指点

谦子

克

这个排序时间复杂度稍微有点难，但只要你静下心来一步一步算，其实也不难

克拿出了笔和纸

克

你想一下堆排的整个过程，第一步建堆，第二步执行N-1次deleteMin()方法,最后取两者复杂度较高的就行了

建堆时间复杂度

克

建堆的时候时间消耗在下沉操作上，而下沉操作最多下沉到底，显然，高度为h的节点下沉代价为O(h)

克

堆中所有元素下沉代价之和就是建堆的代价（时间复杂度）

叶子节点高度为0，下沉代价为0

谦子目不转睛地

克

把堆中不同高度中的所有节点相加起来就是全部的节点

所以问题变为：

① 高度最高多高（高度上界）

② 高度h有多少节点

这里你记住两个结论

有兴趣的可以证证这两个结论

克

这里我们把不同高度的每个节点执行sink所需要的代价累加起来

好复杂啊，头都大了

谦子

克

如果感觉复杂，你最终记住建堆复杂度为O(n)就行了

谦子长舒一口气

N-1次删除复杂度

克

接下来就是N-1次的删除了

n-1次调用deleteMin

deleteMin中包含 swap操作和 sink操作

swap操作代价为常数，sink操作代价为lgn,swap操作相对于sink操作可以忽略不计

则相当于进行了n-1次sink操作

则一共花费的代价为：(n-1)*lgn ~ nlgn

故时间复杂度为O(nlgn)

克

两个步骤相加的复杂度为：O(n)+O(nlgn)，O(nlgn)复杂度高于O(n)，所以堆排序的时间复杂度为O(nlgn)

哦，这样啊，懂了

谦子

稳定性

克

那你说说堆排序是不是稳定的

不是稳定的，就拿5,7,13,5，这个序列来说吧，我用大根堆的结构排序,排序前后两个5的位置会发生变化

谦子

谦子说着说着画了一个图

初始状态的5的先后顺序和排完序的顺序明显不一样了（黄色5跑到红色5左边去了），所以这个排序不是稳定的

谦子

克

恩恩，不错，时间也不早了，早点回去休息吧