leetcode225. 用队列实现栈
题目:
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
- void push(int x) 将元素 x 压入栈顶
- int pop() 移除并返回栈顶元素
- int top() 返回栈顶元素
- boolean empty() 如果栈是空的,返回 true ;否则,返回 false
思路:
Python的deque 是一种双端队列
,适合在两端进行快速插入和删除操作,通过限制它的使用方式来实现队列,进而模拟栈的行为。
解释详见代码~
代码实现(Python):
class MyStack:
'''
双端队列右端(pop、append)相当于栈顶
左端(popleft、appendleft)相当于栈底
模拟队列,双端队列只能右进左出
栈:后进先出
'''
def __init__(self):
'''
初始化双端队列 que ,将用于存储栈的所有元素
'''
self.que=deque()
def push(self, x: int) -> None:
'''
将元素 x 添加到 que 的右端,相当于将元素压入栈顶
'''
self.que.append(x)
def pop(self) -> int:
'''
检查栈是否为空,如果为空则返回 None
将 que 中的所有元素(除了最后一个)依次移到队列的末尾
(双端队列右端)
最后一个元素,即栈顶元素,被移到队列的最前面(双端队列左端)
从 que 中移除并返回该元素。
'''
if self.empty():
return None
for i in range(len(self.que)-1):
self.que.append(self.que.popleft())
return self.que.popleft()
def top(self) -> int:
'''
# 法一:
if self.empty():
return None
return self.que[-1]
# 最右边元素索引为-1或len(self.que)-1
'''
# 法二
'''
检查队列是否为空
将 que 中所有元素(除了最后一个)依次移到队列的末尾
暂存并移除第一个元素(即原队列的右端元素)
将暂存的元素重新添加到队列的末尾
返回暂存的元素。
'''
if self.empty():
return None
for i in range(len(self.que)-1):
self.que.append(self.que.popleft())
temp = self.que.popleft()
self.que.append(temp)
return temp
def empty(self) -> bool:
return not self.que
# 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()
总结:
在这个实现中,双端队列 deque 的右端相当于栈顶。通过 push 方法将元素添加到右端,通过 pop 方法和 top 方法获取并移除右端元素,实现了栈的后进先出(LIFO)特性。