https://leetcode-cn.com/problems/implement-queue-using-stacks-lcci/
实现一个MyQueue类,该类用两个栈来实现一个队列。
示例:
MyQueue queue = new MyQueue(); queue.push(1); queue.push(2); queue.peek(); // 返回 1 queue.pop(); // 返回 1 queue.empty(); // 返回 false
说明:
你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。
peek:查看首个元素,不会移除首个元素,如果队列是空的就返回null
class MyQueue: def __init__(self): self.pushs = [] self.pops = [] def push(self, x: int) -> None: self.pushs.append(x) def pop(self) -> int: if len(self.pops)==0: for i in range(len(self.pushs)): self.pops.append(self.pushs.pop()) return self.pops.pop() def peek(self) -> int: if len(self.pops)==0: for i in range(len(self.pushs)): self.pops.append(self.pushs.pop()) temp = self.pops.pop() self.pops.append(temp) return temp def empty(self) -> bool: if len(self.pushs)==0 and len(self.pops)==0: return True else: return False # Your MyQueue object will be instantiated and called as such: # obj = MyQueue() # obj.push(x) # param_2 = obj.pop() # param_3 = obj.peek() # param_4 = obj.empty()