题目描述
使用栈实现队列的下列操作:
- push(x) — 将一个元素放入队列的尾部。
- pop() — 从队列首部移除元素
- peek() — 返回队列首部的元素
- empty() — 返回队列是否为空
说明:
- 你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top,size 和 is empty 操作是合法的。
- 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列) 来模拟一个栈,只要是标准的栈操作即可。
- 假设所有的操作都是有效的,(例如,一个空的队列不会调用 pop 或 peek 操作)。
思路
栈的顺序为后进先出,而队列的顺序为先进先出。使用两个栈实现队列,一个元素需要经过两个栈才能出队列,在经过第一个栈时元素顺序被反转,经过第二个栈时再次被反转,此时就是先进先出顺序。
参考:https://leetcode-cn.com/problems/implement-queue-using-stacks/solution/shi-yong-liang-ge-zhan-yi-ge-zhuan-men-ru-dui-yi-g/
- 使用两个栈,一个栈(in) 用于元素进栈,一个栈(out) 用于元素出栈。
- pop() 或 peek() 的时候:
- 如果 out 里面有元素,直接从 out 里弹出(pop)或者 peek 元素
- 如果 out 里面没有元素,一次性将 in 里面的所有元素倒入 out。(为此,可以写一个辅助函数 in2out,一次性将 in 里面的元素倒入 out)
注意:一定要保证 out 为空的时候,才能把元素从 in 里拿到 out 中。
代码