'''
Implement the following operations of a stack using queues.
push(x) -- Push element x onto stack.
pop() -- Removes the element on top of the stack.
top() -- Get the top element.
empty() -- Return whether the stack is empty.
Example:
MyStack stack = new MyStack();
stack.push(1);
stack.push(2);
stack.top(); // returns 2
stack.pop(); // returns 2
stack.empty(); // returns false
Notes:
You must use only standard operations of a queue -- which means only push to back, peek/pop from front, size, and is empty operations are valid.
Depending on your language, queue may not be supported natively. You may simulate a queue by using a list or deque (double-ended queue), as long as you use only standard operations of a queue.
You may assume that all operations are valid (for example, no pop or top operations will be called on an empty stack).
'''
class MyStack:
def __init__(self):
"""
Initialize your data structure here.
"""
self.q1 = []
self.q2 = []
def push(self, x: int) -> None:
"""
Push element x onto stack.
"""
if self.q1: self.q1.append(x)
else: self.q2.append(x)
def pop(self) -> int:
"""
Removes the element on top of the stack and returns that element.
"""
if self.q1:
while len(self.q1) > 1:
self.q2.append(self.q1.pop(0))
return self.q1.pop(0)
else:
while len(self.q2) > 1:
self.q1.append(self.q2.pop(0))
return self.q2.pop(0)
def top(self) -> int:
"""
Get the top element.
"""
if self.q1:
while len(self.q1) > 1:
self.q2.append(self.q1.pop(0))
key = self.q1.pop(0)
self.q2.append(key)
return key
else:
while len(self.q2) > 1:
self.q1.append(self.q2.pop(0))
key = self.q2.pop(0)
self.q1.append(key)
return key
def empty(self) -> bool:
"""
Returns whether the stack is empty.
"""
return self.q1 == [] and self.q2 == []
# Your MyStack object will be instantiated and called as such:
# obj = MyStack()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.top()
# param_4 = obj.empty()
# Your MyStack object will be instantiated and called as such:
# obj = MyStack()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.top()
# param_4 = obj.empty()
leetcode 225 Implement Stack using Queues 用队列实现栈 python 双队列
所有Leetcode题目不定期汇总在 Github, 欢迎大家批评指正,讨论交流。'''Implement the following operations of a stack using queues.push(x) -- Push element x onto stack.pop() -- Removes the element on top of the stack.top...