例题描述
使用栈实现队列的下列操作:
push(x)
– 将一个元素放入队列的尾部。
pop()
– 从队列首部移除元素。
top()
– 返回队列首部的元素。
empty()
– 返回队列是否为空。
示例:
MyQueue queue = new MyQueue();
queue.push(1);
queue.push(2);
queue.top(); // 返回 1
queue.pop(); // 返回 1
queue.empty(); // 返回 false
说明:
你只能使用标准的栈操作 – 也就是只有 push to top
, peek/pop from top
, size
, 和 is empty
操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list
或者 deque
(双端队列)来模拟一个栈,只要是标准的栈操作即可。
假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。
解题思路
创建两个栈:一个in
栈,一个out
栈进行操作。
push(x)
入队:直接将元素顺序压入in
栈即可。pop()
出队:将除栈底元素外的所有元素全部按顺序依次压入out
栈,最后将out
栈的栈顶元素弹出即可。top()
返回队列首部的元素:与出队类似,只是最后弹栈完成后,再将元素入栈即可。empty()
返回队列是否为空:判定条件为in
栈与out
栈同时为空,才说明整个队列为空。
图文
假定栈内元素按顺序分别为:9
、5
、2
、7
。
出队
pop
:队列为FIFO型规则(First In First Out),用栈来实现就是先将除栈底元素的元素逐个压入out
栈,最后弹出栈顶元素即可。那么此时此时队列出队的队首元素为9
。
实现方式是:使用临时变量,针对in
栈中元素的循环出栈,每周期保存一次,紧跟着按顺序逐个对out
栈的循环入栈。最后弹出栈顶元素,形成出队行为。
- 注:参数入队条件为
out
栈为空,因为如果out
栈不为空,说明我们已经进行过一次弹栈行为,此时out
中的元素为原队列逆顺序,此时再对in
栈进行参数压栈,就会破坏原来“队列”中的顺序,所以一定要在out
栈为空时才压栈。
in
栈搬运的判定条件为:in
栈不为空。
代码实现
class MyQueue {
public:
stack<int> in;
stack<int> out;
//1. 入队
void push(int x) {
in.push(x);
}
//2. 出队
int pop() {
if(out.empty()){
while(!in.empty()){
int v = in.top();
in.pop();
out.push(v);
}
}
int v = out.top();
out.pop();
return v;
}
//3. 返回队首元素
int top() {
if(out.empty()){
while(!in.empty()){
int v = in.top();
in.pop();
out.push(v);
}
}
int v = out.top();
return v;
}
//4. 返回队列是否为空
bool empty() {
return in.empty() && out.empty();
}
};