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

栈与队列

---

一、用栈实现队列

栈实现先入先出队列，一个输入栈，一个输出栈

class MyQueue {
public:
    //栈实现先入先出队列
    stack<int>output;
    stack<int>input;
    MyQueue() {

    }
    
    void push(int x) {
        input.push(x);
    }
    
    int pop() {
        if (output.empty()) {
            while (!input.empty()) {
                output.push(input.top());
                input.pop();
            }
        }
        int result = output.top();
        output.pop();
        return result;
    }
    
    int peek() {
        int result = pop();
        output.push(result);
        return result;
    }
    
    bool empty() {
        return input.empty() && output.empty();
    }
};

/**
 * Your MyQueue object will be instantiated and called as such:
 * MyQueue* obj = new MyQueue();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->peek();
 * bool param_4 = obj->empty();
 */

二、用队列实现栈

队列实现栈，输入直接输入，输出则将除最后一个元素之外的所有元素移动到另外一个队列中，待输出完毕，将剩余元素返回

class MyStack {
public:
    //使用两个队列实现后入先出的栈
    //输入直接输入
    //输出将出最后一个元素之外的元素都移动到另外一个队列中，完成输出后，再将剩余元素返回
    queue<int>q1;
    queue<int>q2;
    MyStack() {

    }
    
    void push(int x) {
        q1.push(x);
    }
    
    int pop() {
        int len = q1.size();
        for (int i = 0; i < len - 1; i++) {
            q2.push(q1.front());
            q1.pop();
        }
        int result = q1.front();
        q1.pop();
        q1 = q2;
        while (!q2.empty()) {
            q2.pop();
        }
        return result;
    }
    
    int top() {
        int result = this->pop();
        q1.push(result);
        return result;
    }
    
    bool empty() {
        return q1.empty();
    }
};

/**
 * Your MyStack object will be instantiated and called as such:
 * MyStack* obj = new MyStack();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->top();
 * bool param_4 = obj->empty();
 */

---

总结

题目不算太难，养成独立思考的能力，加油！
学习时间90min。
学习资料：《代码随想录》。