请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push
、pop
、peek
、empty
):
实现 MyQueue
类:
void push(int x)
将元素 x 推到队列的末尾int pop()
从队列的开头移除并返回元素int peek()
返回队列开头的元素boolean empty()
如果队列为空,返回true
;否则,返回false
说明:
- 你 只能 使用标准的栈操作 —— 也就是只有
push to top
,peek/pop from top
,size
, 和is empty
操作是合法的。 - 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
示例 1:
输入: ["MyQueue", "push", "push", "peek", "pop", "empty"] [[], [1], [2], [], [], []] 输出: [null, null, null, 1, 1, false] 解释: MyQueue myQueue = new MyQueue(); myQueue.push(1); // queue is: [1] myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue) myQueue.peek(); // return 1 myQueue.pop(); // return 1, queue is [2] myQueue.empty(); // return false
提示:
1 <= x <= 9
- 最多调用
100
次push
、pop
、peek
和empty
- 假设所有操作都是有效的 (例如,一个空的队列不会调用
pop
或者peek
操作)
进阶:
- 你能否实现每个操作均摊时间复杂度为
O(1)
的队列?换句话说,执行n
个操作的总时间复杂度为O(n)
,即使其中一个操作可能花费较长时间。
解法1:栈
队列的特性是 FIFO(先入先出),而栈的特性是 FILO(先入后出)。
知道两者特性之后,我们需要用两个栈来模拟队列的特性,一个栈为入队栈,一个栈为出对栈。
当出队栈存在内容时,出队栈的栈顶,即为第一个出队的元素。
若出队栈无元素,我们的需求又是出队的话,我们就需要将入队栈的内容反序导入出队栈,然后弹出栈顶即可。
注意:根据栈的的特性,我们仅能使用 push 和 pop 操作。
Java:
class MyQueue {
Deque<Integer> inStack;
Deque<Integer> outStack;
public MyQueue() {
inStack = new ArrayDeque<>();
outStack = new ArrayDeque<>();
}
public void push(int x) {
inStack.push(x);
}
public int pop() {
peek();
return outStack.pop();
}
public int peek() {
if (!outStack.isEmpty()) {
return outStack.peek();
}
while (!inStack.isEmpty()) {
outStack.push(inStack.pop());
}
return outStack.peek();
}
public boolean empty() {
return inStack.isEmpty() && outStack.isEmpty();
}
}
/**
* Your MyQueue object will be instantiated and called as such:
* MyQueue obj = new MyQueue();
* obj.push(x);
* int param_2 = obj.pop();
* int param_3 = obj.peek();
* boolean param_4 = obj.empty();
*/
Python3:
class MyQueue:
def __init__(self):
self.inStack = list()
self.outStack = list()
def push(self, x: int) -> None:
self.inStack.append(x)
def pop(self) -> int:
self.peek()
return self.outStack.pop()
def peek(self) -> int:
if self.outStack:
return self.outStack[-1]
while self.inStack:
self.outStack.append(self.inStack.pop())
return self.outStack[-1]
def empty(self) -> bool:
return not self.inStack and not self.outStack
# 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()
复杂度分析
时间的复杂度:
我们使用两个栈,其中栈 inStack 用于入队,另一个栈 outStack 用于出队。
入队时,直接将元素入栈 inStack。时间复杂度 O(1)。
出队时,先判断栈 outStack 是否为空,如果为空,则将栈 inStack 中的元素全部出栈并入栈 outStack,然后再从栈 outStack 中出栈一个元素。如果栈 outStack 不为空,则直接从栈 outStack 中出栈一个元素。均摊时间复杂度 O(1)。
获取队首元素时,先判断栈 outStack 是否为空,如果为空,则将栈 inStack 中的元素全部出栈并入栈 outStack,然后再从栈 outStack 中获取栈顶元素。如果栈 outStack 不为空,则直接从栈 outStack 中获取栈顶元素。均摊时间复杂度 O(1)。
判断队列是否为空时,只要判断两个栈是否都为空即可。时间复杂度 O(1)。
空间复杂度:O(n)。