232.用栈实现队列
使用栈实现队列的下列操作:
push(x) – 将一个元素放入队列的尾部。
pop() – 从队列首部移除元素。
peek() – 返回队列首部的元素。
empty() – 返回队列是否为空。
思路
作为栈与队列周的开胃菜,我选择了这道比较基础且简单的题目。
用栈实现队列,我们要知道其各自对应的特点是什么:
- 栈:先进后出
- 队列:先进先出
外面要用栈实现第一个元素摆在栈顶的位置,这样出来的就符合先进先出。一个栈是解决不了的,我们需要用到两个栈,具体实现思路1如下:
- 建立一个入栈和一个出栈
- push操作,只需将元素push进入栈
- pop操作,首先看出栈是否为空,若为空,则将入栈里头的元素一个个搬过来,不为空则直接输出top()
- peek操作,队列首部即第一个元素,也就是出栈的栈顶,这里可以直接调用上面我们写好的pop函数,由于没有要求弹出,因此我们还要再push回去。
- empty,判断入栈和出栈是否同时为空
代码如下:
class MyQueue {
private:
stack<int> in_queue;
stack<int> out_queue;
public:
MyQueue() {
}
void push(int x) {
in_queue.push(x);
}
int pop() {
if (out_queue.empty()){
while (!in_queue.empty()){
out_queue.push(in_queue.top());
in_queue.pop();
}
}
int res = out_queue.top();out_queue.pop();
return res;
}
int peek() {
int res = this->pop();
out_queue.push(res);
return res;
}
bool empty() {
return in_queue.empty()&&out_queue.empty();
}
};