题 9:用两个栈实现队列
题目:用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead,分别完成在队列尾部插入结点和在队列头部删除结点的功能。——《剑指 Offer》P68
解题思路
队列是先进先出的结构,将一系列的数据入队列后全部出队列,这些数据的顺序不会改变。栈是先进后出的结构,将一系列的数据入栈后全部出栈,这些数据将会逆序出栈。如果将这些逆序的数据再次全部入栈,然后全部出栈就可以恢复这些顺序原有的序列。借助 2 个栈实现队列的关键时,在合适的时候进行出栈操作。例如将 1,2,3 这些数据做入队列操作,也就是将这些数据入第一个栈。
如果要执行出队列操作,由于第一个栈直接出栈会导致顺序逆序,所以要将所有元素出栈进入第二个栈。
此时数据的顺序已经恢复,出队列操作就直接在第二个栈做出栈操作即可。
将数据 4 入队列,也是直接将这些数据入第一个栈就行。
而如果要做出队列操作,注意第二个栈中已经有一些数据了,因此在第二个栈做出栈操作即可。
由此我们可以总结出入队列和出队列操作的时机,入队列操作只需要将数据入第一个栈。出队列操作则需要分类讨论,如果第二个栈有数据,则直接对第二个栈做出栈操作即可。如果第二个栈没有数据,需要先将第一个栈的数据全部出栈后入第二个栈,再对第二个栈做出栈操作。
题解代码
class CQueue:
def __init__(self):
self.stack1 = []
self.stack2 = []
def appendTail(self, value: int) -> None:
self.stack1.append(value)
def deleteHead(self) -> int:
if len(self.stack2) != 0: #第二个栈中有数据
return self.stack2.pop()
elif len(self.stack1) == 0: #队列中没有任何元素
return -1
while len(self.stack1) > 0: #第二个栈中无数据且第一个栈不为空
self.stack2.append(self.stack1.pop())
return self.stack2.pop()
时空复杂度
入队列操作只需要将数据入第一个栈,时间复杂度为 O(1)。
出队列操作时,最坏情况需要将第一个栈中的 n 个元素出栈后进入第二个栈,时间复杂度为 O(n)。
参考资料
《剑指 Offer(第2版)》,何海涛 著,电子工业出版社