两个栈实现队列 牛客网 剑指Offer

两个栈实现队列 牛客网 剑指Offer

 

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

class Solution:
    def __init__(self):
        self.stk1 = []
        self.stk2 = []
         
    def push(self, node):
        self.stk1.append(node)
         
    def pop(self):
        if self.stk2 == []:
            while self.stk1:
                item = self.stk1.pop()
                self.stk2.append(item)
        item = self.stk2.pop()
        return item