zl程序教程

您现在的位置是:首页 >  后端

当前栏目

5、栈和队列——用两个栈实现队列(python版)

Python队列队列 实现 两个
2023-09-11 14:20:01 时间

《剑指Offer》

题目描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。


栈:先进后出;队列:先进先出

若要想用两个栈实现队列的过程,可根据栈的特性,即先进后出,让一个栈用来进行存数,计为1号栈,可实现“入队”操作。

当想实现出队操作时,需要实现的效果是把1号栈中最开始存的数,即栈底元素先出,但我们发现当该栈进行取数据时,栈底元素只能最后出来,这时候就可以用另一个栈,计为2号栈,来将1号栈取出的数依次按顺序再存进来,1号栈中的栈底元素就变成了2号栈中的栈顶元素,当从2号栈中进行取数时,就实现了“出队操作”。

class Solution:
    def __init__(self):
        self.acceptStack = []
        self.outputStack = []
    def push(self, node):
        # write code here
        self.acceptStack.append(node)
    def pop(self):
        # return xx
        if self.outputStack == []:
            while self.acceptStack:
                self.outputStack.append(self.acceptStack.pop())
        if self.outputStack != []:
            return self.outputStack.pop()
        else:
            return None
# 可以测试一下
solution = Solution()
solution.push(123)
solution.pop()

总结:由于栈的特殊操作性质,当对元素进行两次的存取时,就相当于实现了一个“逆置换“。