用队列实现栈
概述:请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。
输入:
["MyStack", "push", "push", "top", "pop", "empty"]
[[], [1], [2], [], [], []]
输出:
[null, null, null, 2, 2, false]
方法一:两个队列
思路:首先定义两个空队列,一个用于存储,一个用于弹出。每次入队结束以后两两交换即可。
# 两个队列
# 首先定义两个空队列,一个用于存储,一个用于弹出。
# 每次入队结束以后两两交换即可。
class MyStack:
def __init__(self):
self.queue1 = collections.deque()
self.queue2 = collections.deque()
def push(self, x: int) -> None:
self.queue2.append(x)
while self.queue1:
self.queue2.append(self.queue1.popleft())
self.queue1, self.queue2 = self.queue2, self.queue1
def pop(self) -> int:
return self.queue1.popleft()
def top(self) -> int:
return self.queue1[0]
def empty(self) -> bool:
return not self.queue1
方法二:一个队列
思路:此算法思路非常简单,入队一个弹出一个即可。
# 一个队列
# 此算法思路非常简单,入队一个弹出一个即可。
class MyStack:
def __init__(self):
self.queue = collections.deque()
def push(self, x: int) -> None:
n = len(self.queue)
self.queue.append(x)
for i in range(n):
self.queue.append(self.queue.popleft())
def pop(self) -> int:
return self.queue.popleft()
def top(self) -> int:
return self.queue[0]
def empty(self) -> bool:
return not self.queue
总结
我怎么感觉两个队列的要难理解,命名多此一举,用一个队列就可以解决的事啊!