2021-10-06 leetcode 数据结构 队列与栈 232.用栈实现队列 225.用队列实现栈

232.用栈实现队列

双栈实现
利用两个栈，改变信息存放形式，以得到LIFO下的数据

class MyQueue {
private:
    stack<int> s1, s2;
    int n, ans;
public:
    MyQueue() {

    }
    
    void push(int x) {
        s1.push(x);
    }
    
    int pop() {
        n = s1.size();
    //按队列顺序放入s2
    for(int i = 0; i < n; i++){
        s2.push(s1.top());
        s1.pop();
    }
    ans = s2.top();
    s2.pop();
    //元素拿出后再将栈装回去
    n = s2.size();
    for(int i = 0; i < n; i++){
        s1.push(s2.top());
        s2.pop();
    }
    return ans;
    }
    
    int peek() {
    n = s1.size();
    //按队列顺序放入s2
    for(int i = 0; i < n; i++){
        s2.push(s1.top());
        s1.pop();
    }
    ans = s2.top();
    //元素拿出后再将栈装回去
    for(int i = 0; i < n; i++){
        s1.push(s2.top());
        s2.pop();
    }
    return ans;
    }
    
    bool empty() {
        if(s1.empty())
            return true;
        return false;
    }
};

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

225.用队列实现栈

双队列实现栈
利用两个队列，使得移除的信息可暂时存放在另一个队列中，得到所求值后，再依次存入。

class MyStack {
private:
    queue<int>q1, q2;
    int ans, n;
public:
    MyStack() {

    }
    
    void push(int x) {
        q1.push(x);
    }
    
    int pop() {
        //要取最后一个，则将其他元素暂时存入另一个队列中
        n = q1.size();
        for(int i = 0; i < n-1; i++){
            q2.push(q1.front());
            q1.pop();
        }
        ans = q1.front();
        q1.pop();
        //装回去
        n = q2.size();
        for(int i = 0; i < n; i++){
            q1.push(q2.front());
            q2.pop();
        }
        return ans;
    }
    
    int top() {
        //要取最后一个，则将其他元素暂时存入另一个队列中
        n = q1.size();
        for(int i = 0; i < n-1; i++){
            q2.push(q1.front());
            q1.pop();
        }
        ans = q1.front();
        q2.push(ans);
        q1.pop();//注意将q1清空
        //装回去
        n = q2.size();
        for(int i = 0; i < n; i++){
            q1.push(q2.front());
            q2.pop();
        }
        return ans;
    }
    
    bool empty() {
        if(q1.empty())
            return true;
        return false;
    }
};

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