代码随想录算法训练营第十天 | LeetCode232 用栈实现队列 LeetCode225 用队列实现栈

LeetCode232 用栈实现队列

题目链接：用栈实现队列

文章链接：用栈实现队列

视频链接：用栈实现队列

思路

这道题是用栈实现队列，栈的特点是先进后出，进出都是在一端完成；而队列是先进先出，进出是在两端完成的，所有要想用栈实现队列，一个栈肯定是不够的，所以这里要用到两个栈，分别作为进栈和出栈，模仿队列的样子。

队列的入队实现：进“进栈“

队列的出队实现：先判断”出栈“是否为空，为空时，让”进栈“里的元素不断出栈去”出栈“，”出栈“的栈顶元素就是出队的元素。

取队头元素的实现：代码逻辑与pop（）相同，但这里是用函数复用，而不是cv大法，利用this指针，这样代码健壮性更好。

代码

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

    }

    void push(int x) {
        stIn.push(x);
    }

    int pop() {
        if (stOut.empty())
        {
            while (!stIn.empty())
            {
                stOut.push(stIn.top());
                stIn.pop();
            }
        }
        int result = stOut.top();
        stOut.pop();
        return result;
    }

    int peek() {
        int res = this->pop();
        stOut.push(res);
        return res;
    }

    bool empty() {
        if (stIn.empty() && stOut.empty())
            return true;
        else
            return false;
    }
};

LeetCode225 用队列实现栈

题目链接：用队列实现栈

文章链接：用队列实现栈

视频链接：用队列实现栈

思路

队列是先进先出，栈是先进后出，所以出栈的元素一定是队列中最后一个元素，这里需要用到两个队列，但这并不是和用栈实现队列

这题完全一样，这里的两个队列不是一个是输出队列一个是输入队列，而是que1用来存储栈的元素，que2用来拷贝que1最后一个元素之前的所有元素，因为出栈的元素就是que1最后的元素，所以要把que1前面的那些元素拷贝到que2里，等出栈操作完成之后再将que2的元素拷贝回que1。

进栈操作等价于入队que1。

代码

#include<iostream>
#include<queue>

using namespace std;

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

    }

    void push(int x) {
        que1.push(x);
    }

    int pop() {
        int size = que1.size();
        size--;
        while (size--)
        {
            que2.push(que1.front());
            que1.pop();
        }

        int result = que1.front();
        que1.pop();
        que1 = que2;
        while (!que2.empty())
        {
            que2.pop();
        }
        return result;
    }

    int top() {
        return que1.back();
    }

    bool empty() {
        return que1.empty();
    }
};