最小栈和最小队列

最新推荐文章于 2024-06-08 11:45:57 发布

360技术

最新推荐文章于 2024-06-08 11:45:57 发布

阅读量784

点赞数 2

原文链接：https://www.huodongxing.com/event/5500608998700

版权

奇技指南

本文来自奇舞周刊，作者高峰，360奇舞团前端工程师，W3C性能工作组/WOT工作组成员。

相信栈和队列这两种数据结构大家一定都不陌生吧。这两种数据结构有个特殊形式：即最小栈和最小队列。

下面我们分别来介绍下这两个带有“特异功能”的数据结构。

最小栈

什么是最小栈

实现一个栈，带有出栈（pop），入栈（push），取最小元素（getMin）三个方法。并且这三个方法的时间复杂度都是O（1）。

设计思路

看了上面对于最小栈的定义后，可能大部分人都能够快速想到这样的思路：

创建一个栈，再额外声明一个变量，用于存储栈的最小值。将第一个入栈的元素赋值给最小值变量。之后在每次进行入栈操作时，判断最小值变量与新入栈元素的大小。如果新入栈元素小于最小值变量，则修改最小值变量的值为新入栈元素。

这样我们就实现了一个栈，并且还会有一个变量指向栈的最小值。

不过遗憾的是，这个思路是错误的。因为当我们进行出栈操作的时候，如果最小值出栈了，那所声明的最小值变量就没有意义了。

所以我们在最小值出栈的时候，需要有更换顶替当前最小值的次最小值。

上面说了，我们在最小值出栈时，需要有更换顶替当前最小值的值。我们可以通过两个栈来实现，第一个栈用于存放入栈的元素，第二个栈用于存放最小元素。

具体思路如下：

注意：B栈中之所以存索引，是因为这样可以规避一些问题。如当后加入的值和最小值相同时，如果我们不重复入栈B，那么出栈的时候对于B栈的判断就可能出问题（后入的等值元素可能会导致最小值提前出栈）。存索引就可以避免重复入栈最小值这个问题，节省一定的空间。

算法实现

为了便于展示，笔者使用Vue实现了这一算法。详情可以参见https://jsbin.com/lokedoxere。实现效果如下图：

640?wx_fmt=png

算法逻辑如下：

至此，我们实现了最小栈的逻辑。其实最小队列的实现思路与最小栈类似。不过又略有点不同。下面我们一起来看看吧。

最小队列的实现

什么是最小队列

实现一个队列，带有出队（deQueue），入队（enQueue），取最小元素（getMin）三个方法。要保证这三个方法的时间复杂度都尽可能小。

设计思路

这里我就不卖关子了，直接进入正题。

我们可以通过两个队列实现，第一个队列用于存放入队的元素，第二个队列用于存放最小值。

详细思路如下：

创建两个队列分别为A和B，A用于存放入队的元素，B用于存放最小值（严格来说，B不能称作队列，它会将后入的元素删除）。
当第一个元素入队A时，同时将其入队B，因为这是就唯一一个值，所以是最小值。
之后每个元素入队A时，从B队尾开始比较，如果当前值大于等于（要有等于）B队尾值，则入队；否则，则删除B队尾值；然后再次比对B队尾值和当前值大小，直到当前值大于等于（要有等于） B队尾值或B为空，则当前值入队B尾。
这时每次B队中队首元素即为最小值。
当A中的元素出队时，如果元素值刚好与B队首元素相等，则B队首的元素也出队。（第三步中包含等于的判断，就是为了避免出队时等值元素不会出问题。）
这时deQueue/getMin时间复杂度为O(1)。enQueue为O(n)。