【剑指offer】 面试题9 用两个栈实现队列

题目要求

用两个栈实现一个队列。队列的声明如下，请实现它的两个成员函数，分别完成在队列尾部插入节点和在队列的头部删除节点

class solution
{
public:
    void push(int node);
    int pop();

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

解题思路

用两个先进后出的栈，实现一个先进先出的队列。
举个例子，我们先输入a,b,c，把他们放到第一个栈中，然后根据队列的性质取出首元素a,但是现在在栈中，栈顶元素是c,无法取出a,那又怎么办呢？这时候，我们想到可以把第一个栈中的元素依次再放到第二个栈中，此时栈2中的元素为a,b,c其中a在栈顶位置，那么即可实现队列首元素删除操作。

总结要点：进栈，插入元素统一放到第一个栈中，需要删除操作时，判断第二个栈是不是空，非空的话直接进行取栈顶，进行删除；如为空，那么把第一个栈的元素放到第二个栈再进行删除操作。

主要代码 c++

class Solution
{
public:
    void push(int node) {
        stack1.push(node); // 插入都用stack1
    }
    int pop() {
        if(stack2.size()<=0) // 删除操作借助stack2 二次进栈达到队列的效果
        {
            while(!stack1.empty())
            {
                int tmp = stack1.top();
                stack1.pop();
                stack2.push(tmp);
            }
        }
        int data = stack2.top(); // 如果stack2非空 那么直接删除即可
        stack2.pop();
        return data;
    }

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

扩展

用两个队列实现栈的功能就会稍微复杂一些，对于每一次的栈删除操作，都需要俩个队列相互配合，利用空队列头部进行删除，可以思考一下。