day13-232.用栈实现队列

232.用栈实现队列

力扣题目链接(opens new window)

使用栈实现队列的下列操作：

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();
    }
};