用两个栈实现队列
题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
# -*- coding:utf-8 -*- class Solution: def __init__(self): self.stack1 = [] self.stack2 = [] def push(self, node): # write code here self.stack1.append(node) def pop(self): # return xx if self.stack2: return self.stack2.pop() elif self.stack1 == []: return None else: while self.stack1: self.stack2.append(self.stack1.pop()) return self.stack2.pop()
嗯~题没读透~不知道应该返回何种数据类型~故记录一下