两个栈实现队列 牛客网 剑指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