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

你可能感兴趣的:(算法,算法)