剑指offer——07用两个栈实现队列

题目描述

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

 

题解：

　　有两个栈，stack1和stack2

　　push只在stack1上操作，pop只在stack2上操作

　　当stack2不为空时，直接从stack2栈顶弹出

　　当stack2为空时，将stack1的数据全部弹出压入stack2中，然后再从stack2中弹出。

 

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

    int pop() {
        if (stack2.empty())
        {
            while (!stack1.empty())
            {
                stack2.push(stack1.top());
                stack1.pop();
            }            
        }
        int res = 0;
        if (!stack2.empty())
        {
            res = stack2.top();
            stack2.pop();
        }
        return res;
    }

private:
    stack<int> stack1;
    stack<int> stack2;
};