牛客网-用两个栈实现队列

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

 

 

解题思路

class Solution
{
public:
    void push(int node) {
        stk1.push(node);
    }

    int pop() {
        int ans = -1;
        if(stk2.empty()){
            while(!stk1.empty()){
                stk2.push(stk1.top());
                stk1.pop();
            }
        }
        if(stk2.empty()) return ans;
        ans = stk2.top();
        stk2.pop();
        return ans;
    }

private:
    stack<int> stk1;
    stack<int> stk2;
};