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

题目描述：

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

本题知识点：

队列  栈

解题思路：

栈的性质为先进后出，队列为先进先出。

模拟出队列。

stack1 = {a, b, c}, stack2 = { }

首先根据队列的先进先出规则，应该先弹出stack1中的第一个元素，需要将stack1中的所有元素push到stack2中。

stack1 = { }, stack2 = {c, b, a}

此时将stack2中的顶元素pop，返回a。其次将stack2中的元素b按顺序pop，返回b。

stack1 = { }, stack2 = {c}

此时若要push一个元素d，则直接push到stack1中即可。

stack1 = {d}, stack2 = {c}

若需要继续pop，则直接弹出stack2中的元素直到stack2为空。

通过上述可以得出，在队列push时，直接将元素push到stack1中即可。在队列pop时，当stack2为空且stack1不为空时，将stack1中的元素按顺序push到stack2中并弹出，此时队列pop的元素返回值为stack2的顶元素。

代码：

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

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

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