LeetCode 232. 用栈实现队列

LeetCode 232. 用栈实现队列

思路:

q2用作辅助。

代码：

class MyQueue {
public:
    MyQueue() {

    }
    
    void push(int x) {
        q1.push(x);
    }
    
    int pop() {
        while(q1.size() > 1){
            q2.push(q1.top());
            q1.pop();
        }
        int res = q1.top();
        q1.pop();
        while(q2.size()){
            q1.push(q2.top());
            q2.pop();
        }
        return res;
    }
    
    int peek() {
        while(q1.size() > 1){
            q2.push(q1.top());
            q1.pop();
        }
        int res = q1.top();
        q2.push(res);
        q1.pop();
        while(q2.size()){
            q1.push(q2.top());
            q2.pop();
        }
        return res;
    }
    
    bool empty() {
        return q1.empty();
    }

    stack<int> q1,q2;
};

/**
 * 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();
 */