Day09:232.用栈实现队列、225. 用队列实现栈

最新推荐文章于 2024-10-20 15:47:10 发布

鱼丸不灵

最新推荐文章于 2024-10-20 15:47:10 发布

阅读量53

点赞数

文章标签： leetcode

本文链接：https://blog.csdn.net/W2575863886/article/details/133950135

版权

文章目录

前言

首先要清楚栈和队列的实现方法。

232.用栈实现队列

题目链接

思路

题目需要自己写四个函数，分别是push、pop、peek、empty，其中重点讲pop和peek。

pop
因为栈是后进先出，所以应该使用两个栈，一个用来push被pop的元素，让栈内元素顺序全部倒转后，就可以pop达到和队列一样先进先出的效果。注意应该先判断stackOut内是不是空的，只有栈为空时才可以存元素。
peek
实现原理其实和pop一样，但是它可以使用类里已经写好的pop函数，重点是如何使用呢？第一种办法就是重新再写一遍，但这种方法在大型工业开发里很难维护，容易出bug。所以我们采用第二种方法：this指针调用pop函数。
empty
如果进栈和出栈都为空的话，说明模拟的队列为空了。

代码实现

class MyQueue {
public:
    stack<int> stIn;
    stack<int> stOut;
    /** Initialize your data structure here. */
    MyQueue() {

    }
    /** Push element x to the back of queue. */
    void push(int x) {
        stIn.push(x);
    }

    /** Removes the element from in front of queue and returns that element. */
    int pop() {
        // 只有当stOut为空的时候，再从stIn里导入数据（导入stIn全部数据）
        if (stOut.empty()) {
            // 从stIn导入数据直到stIn为空
            while(!stIn.empty()) {
                stOut.push(stIn.top());
                stIn.pop();
            }
        }
        int result = stOut.top();
        stOut.pop();
        return result;
    }

    /** Get the front element. */
    int peek() {
        int res = this->pop(); // 直接使用已有的pop函数
        stOut.push(res); // 因为pop函数弹出了元素res，所以再添加回去
        return res;
    }

    /** Returns whether the queue is empty. */
    bool empty() {
        return stIn.empty() && stOut.empty();
    }
};

时间复杂度: push和empty为O(1), pop和peek为O(n)
空间复杂度: O(n)

总结

一定要懂得复用，功能相近的函数要抽象出来，不要大量的复制粘贴，很容易出问题！（踩过坑的人自然懂），工作中如果发现某一个功能自己要经常用，同事们可能也会用到，自己就花点时间把这个功能抽象成一个好用的函数或者工具类，不仅自己方便，也方便了同事们。

225. 用队列实现栈

题目链接

思路

题目要求实现以下四种功能：
push(x) – 元素 x 入栈
pop() – 移除栈顶元素
top() – 获取栈顶元素
empty() – 返回栈是否为空

pop
队列的pop实现可以只使用一个queue就达到目的，可以将目标元素之前的元素都出队再插入队尾，这样就达到栈的目的。但在这我犯了一个错误，把pop内的while循环条件写成了while(!queue.empty( )||size–)，这样函数执行结果超时，因为queue.empty( )是一直满足的，所以要写写应该是（!queue.empty( )&&size–)。

代码实现

class MyStack {
public:
    queue<int> que;
    /** Initialize your data structure here. */
    MyStack() {

    }
    /** Push element x onto stack. */
    void push(int x) {
        que.push(x);
    }
    /** Removes the element on top of the stack and returns that element. */
    int pop() {
        int size = que.size();
        size--;
        while (size--) { // 将队列头部的元素（除了最后一个元素外） 重新添加到队列尾部
            que.push(que.front());
            que.pop();
        }
        int result = que.front(); // 此时弹出的元素顺序就是栈的顺序了
        que.pop();
        return result;
    }

    /** Get the top element. */
    int top() {
        return que.back();
    }

    /** Returns whether the stack is empty. */
    bool empty() {
        return que.empty();
    }
};