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

232.用栈实现队列

先看代码：

class MyQueue {
public:
    stack<int> a,b;
    MyQueue() {

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

    }
    
    int pop() {
        while(a.size() > 1)
        {
            b.push(a.top());
            a.pop();
        }
        int t = a.top();
        a.pop();
        while(b.size() > 0)
        {
            a.push(b.top());
            b.pop();
        }
        return t;
    }
    int peek() {
        while(a.size() > 1)
        {
            b.push(a.top());
            a.pop();
        }
        int t = a.top();
        while(b.size() > 0)
        {
            a.push(b.top());
            b.pop();
        }
        return t;
    }
    bool empty() {
        return a.empty();
    }
};

我是一看就会一写就费同学，栈和队列对于学习计算机的同学来说应该不陌生，甚至对于计算机考研的人来说很熟悉，对于学习C++STL的人来说的人来说更加熟悉，队列和栈的底层数据结构都是双端队列。
栈就是遵循FILO（First In Last Out） 的原则，先进后出，这里是用栈实现队列，一个栈是做不到的，只能两个栈，或者一个栈再搭配一个其他数据结构，这里就是让用栈，所以用两个栈，其中pop和peek，相比较是复杂一点，总之，目的就一个，就是获取栈底元素，pop获取后就将该元素弹出，peek就是单纯的获取，然后其他元素各自归位。

225. 用队列实现栈

先看代码：

class MyStack {
public:
    queue<int> a,b;
    MyStack() {

    }
    void push(int x) {
        a.push(x);
    }
    
    int pop() {
        while(a.size() > 1)
        {
            b.push(a.front());
            a.pop();
        }
        int t = a.front();
        a.pop();
        while(b.size() > 0)
        {
            a.push(b.front());
            b.pop();
        }
        return t;
    }
    
    int top() {
        while(a.size() > 1)
        {
            b.push(a.front());
            a.pop();
        }
        int t = a.front();
        b.push(a.front());
        a.pop();
        while(b.size() > 0)
        {
            a.push(b.front());
            b.pop();
        }
        return t;
    }
    
    bool empty() {
        return a.empty();
    }
};

这个使用队列实现栈，同样一个队列也是完不成的，所以需要两个队列。队列是FIFO（First In First Out）先进先出，所以再pop和peek上有点区别，pop是从头出，stack的pop是输出尾部，所以先将队列尾部前的元素转移到另一个队列里，然后弹出，再将剩余的元素归位，peek是获取到最后一个元素后，最后一个元素也要转移到另一个队列里，再整体归位。