题目:用两个栈来实现一个队列,完成队列的出栈和入栈操作。队列中的元素为int类型。
思路:
栈为先进后出,队列为先进先出。所以对于一列值,在A栈里先进去的后面出,让A栈中的元素出来后再进入B栈,则从B栈出去的时候就和进入A栈的顺序一致。所以就是让元素从A栈进入,然后出来进入B栈,从B栈出就可以实现整体的先进先出的队列操作。
python用到的数据结构:list
list.pop([index])——可以实现对列表中对应索引处的值的删除,并将值返回,默认情况下删除并返回最后一个元素
代码:
class Solution:
def init(self):
self.stackIn = []
self.stackOut = []
def push(self, node):
self.stackIn.append(node)
def pop(self):
if not self.stackOut:
while self.stackIn:
self.stackOut.append(self.stackIn.pop(-1))
return self.stackOut.pop(-1)
if name == ‘main’:
solution = Solution()
solution.push(1)
solution.push(2)
solution.push(3)
solution.pop()
solution.pop()
solution.push(4)
solution.pop()
solution.push(5)
solution.pop()
代码复制链接:https://blog.csdn.net/u010005281/article/details/79811287