代码随想录算法训练营day8（栈与队列）

用栈实现队列

• https://leetcode.cn/problems/implement-queue-using-stacks/description/

push(x) -- 将一个元素放入队列的尾部。
pop() -- 从队列首部移除元素。
peek() -- 返回队列首部的元素。
empty() -- 返回队列是否为空。

思路

• 初始化两个栈，一个in栈，一个out栈
• push()时，直接将元素插入in栈即可
• pop()时，如果out栈为空，要将in栈中的所有元素导入out栈；如果out栈不为空，将元素弹出即可
• peek()同pop()一样，但是不用弹出
• 如果in栈和out栈都为空，则说明队列为空

class MyQueue:def __init__(self):self.stack_in = []self.stack_out = []def push(self, x: int) -> None:self.stack_in.append(x)def pop(self) -> int:if self.empty():return Noneif self.stack_out:return self.stack_out.pop()else:while self.stack_in:tmp = self.stack_in.pop()self.stack_out.append(tmp)return self.stack_out.pop()def peek(self) -> int:if self.empty():return Noneif self.stack_out:return self.stack_out[-1]else:while self.stack_in:tmp = self.stack_in.pop()self.stack_out.append(tmp)return self.stack_out[-1]def empty(self) -> bool:if not self.stack_in and not self.stack_out:return Truereturn 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()