代码随想录算法训练营第十天｜ 232.用栈实现队列 225. 用队列实现栈

232.用栈实现队列

力扣

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 result = this -> pop();
        stOut.push(result);
        return result;
    }
    
    bool empty() {
        return stIn.empty() && stOut.empty();
    }
};

使用两个相反的栈

当输出的栈是空的时候才能输入，否则栈的顺序会改变

把开始输入的栈的顶部的元素输入到输出的栈，再把输出的栈的元素全部输出

peek()的实现，直接复用了pop()，然后再把pop的值push即可

225.用队列实现栈

力扣

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

    }
    
    void push(int x) {
        que.push(x);
    }
    
    int pop() {
        int size = que.size();
        size --;
        while(size --){
            que.push(que.front());
            que.pop();
        }
        int result = que.front();
        que.pop();
        return result;
    }
    
    int top() {
        return que.back();
    }
    
    bool empty() {
        return que.empty();
    }
};

实现栈只需要一个队列就行，把队列的前n-1个值pop然后再push就能把最后一个放在第一位

然后一个一个输出就行