第一题 很经典的栈模拟队列 https://leetcode.cn/problems/implement-queue-using-stacks/submissions/693619507/
这道题倒是没啥算法性质,主要还是模拟(话说模拟也不简单啊,挺考验思维能力的),一个栈的话只能先进后出所以怎么都无法实现先进先出的效果,所以肯定是两个栈,一个模拟入队,一个模拟出队,具体实现时是一个栈作为压入栈,在压入数据时只往这个栈中压入,记为In_stack;另一个栈只作为弹出栈,在弹出数据时只从这个栈弹出,记为out_stack。因为数据压入栈的时候,顺序是先进后出的。那么只要把In_stack的数据再压入out_stack中,顺序就变回来了,这听上去简单但是有两个要注意:
一次性全部压入:从
in_stack向out_stack转移时,必须一次性转移所有元素。非空不压入:只要
out_stack不为空,就绝对不能进行转移。
违反1的情况举例:1~5依次压入in_stack,in_stack的栈顶到栈底为5~1,从in_stack压入out_stack时,只将5和4压入了out_stack,out_stack还剩下1、2、3没有压入。此时如果用户想进行弹出操作,那么4将最先弹出,与预想的队列顺序就不一致。
违反2的情况举例:1~5依次压入in_stack,in_stack将所有的数据压入out_stack,此时从out_stack的栈顶到栈底就变成了1~5。此时又有6~10依次压入in_stack,out_stack不为空,in_stack不能向其中压入数据。如果违反2压入了out_stack,从out_stack的栈顶到栈底就变成了6~10、1~5。那么此时如果用户想进行弹出操作,6将最先弹出,与预想的队列顺序就不一致.
class MyQueue: def __init__(self): self.in_stack = [] self.out_stack = [] def push(self, x: int) -> None: self.in_stack.append(x) def pop(self) -> int: self._help() return self.out_stack.pop() def peek(self) -> int: self._help() return self.out_stack[-1] def empty(self) -> bool: return not self.in_stack and not self.out_stack def _help(self): if not self.out_stack: with self.in_stack: self.out_stack.append(self.in_stack.pop())def _help(self):
# 规则2:只有out_stack为空时才转移
if not self.out_stack:
# 规则1:一次性转移in_stack中的所有元素
while self.in_stack: # 这个while循环保证了"一次性全部"
self.out_stack.append(self.in_stack.pop())