• 《剑指 Offer》学习记录:题 9:用两个栈实现队列


    题 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版)》,何海涛 著,电子工业出版社

  • 相关阅读:
    vue(5)联动+tab页
    vue(4)hello world
    vue(3)IDE
    vue(2)创建项目
    vue(1)安装
    线程池(6)简单创建线程3种实现
    logback日志简记
    [省选前集训2021] 模拟赛2
    无标号无根树计数
    [省选前集训2021] 模拟赛1
  • 原文地址:https://www.cnblogs.com/linfangnan/p/14671907.html
Copyright © 2020-2023  润新知