【剑指offer】Q9_用两个栈实现队列

题目

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

解答

两个栈来回倒就能访问栈底了，没啥说的

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stark1 = []  # 用list模拟栈 在后续代码中不得违反栈的操作规则
        self.stark2 = []  # 即只能对其 append 或 pop

    def push(self, node):
        # 搬移元素
        for _ in range(len(self.stark2)):
            self.stark1.append(self.stark2.pop())
        # 插入元素
        self.stark1.append(node)

    def pop(self):
        # 搬移元素
        for _ in range(len(self.stark1)):
            self.stark2.append(self.stark1.pop())
        # 弹出元素
        return self.stark2.pop()


s = Solution()
# ["PSH1","PSH2","PSH3","POP","POP","PSH4","POP","PSH5","POP","POP"]
s.push(1)
s.push(2)
s.push(3)
print(s.pop())
print(s.pop())
s.push(4)
print(s.pop())
s.push(5)
print(s.pop())
print(s.pop())
# 对应输出应该为: 1,2,3,4,5