代码随想录算法训练营day10 | 232. 用栈实现队列、225. 用队列实现栈

第一题

链接: 用栈实现队列

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：

实现 MyQueue 类：

void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空，返回 true ；否则，返回 false
说明：

• 你 只能 使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。

• 你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。

思路

使用两个栈来实现，一个不行。定义两个栈，一个进栈，一个出栈；一个用来模拟进队列，一个用来模拟出队列。

解题方法

在push数据的时候，只要数据放进输入栈就好，但在pop的时候，操作复杂一些，输出栈如果为空，就把进栈数据全部导入进来，再从出栈弹出数据，如果输出栈不为空，则直接从出栈弹出数据即可。

最后如何判断队列为空呢？如果进栈和出栈都为空的话，说明模拟的队列为空了。

复杂度

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

Code

class MyQueue {
public:
    stack<int> stIn;
    stack<int> stOut;

    MyQueue() {}

    // 将元素 x 推到队列的末尾
    void push(int x) { 
        stIn.push(x); 
    }

    // 从队列的开头移除并返回元素
    int pop() {
        // 只有当stOut为空的时候，再从stIn里导入数据（导入stIn全部数据）
        if (stOut.empty()) {
            // 从stIn导入数据直到stIn为空
            while (!stIn.empty()) {
                stOut.push(stIn.top());
                stIn.pop();
            }
        }
        int res = stOut.top();
        stOut.pop();
        return res;
    }

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

    // 如果队列为空，返回 true ；否则，返回 false
    bool empty() { 
        return stIn.empty() && stOut.empty(); 
        }
};

用队列实现栈

链接: 用队列实现栈

请你仅使用两个队列实现一个后入先出（LIFO）的栈，并支持普通栈的全部四种操作（push、top、pop 和 empty）。

实现 MyStack 类：

void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的，返回 true ；否则，返回 false 。

注意：

• 你只能使用队列的基本操作 —— 也就是 push to back、peek/pop from front、size 和 is empty 这些操作。

• 你所使用的语言也许不支持队列。 你可以使用 list （列表）或者 deque（双端队列）来模拟一个队列 , 只要是标准的队列操作即可。

思路

前提：这是一个单向队列
根据上题，想着用一个输入队列，一个输出队列，就可以模拟栈的功能，此时是不行的。
此时有两种思路：

• 使用两个队列
• 使用一个队列

解题方法

使用两个队列

用两个队列que1和que2实现队列的功能，把que1最后面的元素以外的元素都备份到que2，然后弹
出最后面的元素，再把其他元素从que2导回que1。

细节看代码注释。

使用一个队列

一个队列在模拟栈弹出元素的时候只要将队列头部的元素（除了最后一个元素外） 重新添加到队列尾部，此时再去弹出元素就是栈的顺序了。

复杂度(两种方法复杂度一样)

• 时间复杂度：pop为O(n)，其他为O(1)
• 空间复杂度：O(n)

Code

使用两个队列

class MyStack {
public:
    queue<int> que1;
    queue<int> que2;
    MyStack() {

    }
    
    //将元素 x 压入栈顶
    void push(int x) {
        que1.push(x);
    }
    
    // 移除并返回栈顶元素
    int pop() {
        int size = que1.size();
        size--;
        while(size--) { // 将que1 导入que2，但要留下最后一个元素
            que2.push(que1.front());
            que1.pop();
        }

        int res = que1.front(); // 留下的最后一个元素就是要返回的值
        que1.pop();
        que1 = que2; // 再将que2赋值给que1
        while(!que2.empty()) { // 清空que2
            que2.pop();
        }
        return res;
    }
    
    //返回栈顶元素
    int top() {
        return que1.back();
    }
    
    //如果栈是空的，返回 true ；否则，返回 false 。
    bool empty() {
        return que1.empty();
    }
};

使用一个队列

class MyStack {
public:
    queue<int> que;

    MyStack() {}

    // 将元素 x 压入栈顶
    void push(int x) { que.push(x); }

    // 移除并返回栈顶元素
    int pop() {
        int size = que.size();
        size--;
        while (size--) { // 将队列头部的元素（除了最后一个元素外）
                         // 重新添加到队列尾部
            que.push(que.front());
            que.pop();
        }

        int res = que.front(); // 此时弹出的元素顺序就是栈的顺序
        que.pop();
        return res;
    }

    // 返回栈顶元素
    int top() { 
        return que.back(); 
        }

    // 如果栈是空的，返回 true ；否则，返回 false 。
    bool empty() { 
        return que.empty(); 
        }
};

总结

• 今天的题目比较简单，看文字看不懂可以去找视频看一下，只要懂的这两种STL的特性就很简单了。
• 参考文档
  链接: 用栈实现队列
  链接: 用队列实现栈