剑指Offer(第2版)——面试题9：用两个栈实现队列

题目：

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

1.解题思路：

• 这题的意图是创建两个栈stack1和stack2，使用两个“先进后出”的栈实现一个“先进先出”的队列。
• 我们通过一个具体的例子分析往该队列插入和删除元素的过程。首先插入一个元素a，不妨先把它插入到stack1，此时stack1中的元素有{a}，stack2为空。再压入两个元素b和c，还是插入到stack1中，此时stack1的元素有{a,b,c}，其中c位于栈顶，而stack2仍然是空的，如图(a)所示。
• 这个时候我们试着从队列中删除一个元素。按照先入先出的规则，由于a比b、c先插入队列中，最先删除的元素应该是a。元素a存储在stack1中，但并不在栈顶，因此不能直接进行删除操作。注意stack2我们一直没有使用过，现在是让stack2发挥作用的时候了。如果我们把stack1中的元素逐个弹出压入stack2，元素在stack2中的顺序正好和原来在stack1中的顺序相反。因此经过3次弹出stack1和要入stack2操作之后，stack1为空，而stack2中的元素是{c,b,a}，这个时候就可以弹出stack2的栈顶a了。此时的stack1为空，而stack2的元素为{b,a}，其中b在栈顶，如图(b)所示。
• 因此我们的思路是：当stack2中不为空时，在stack2中的栈顶元素是最先进入队列的元素，可以弹出；如果stack2为空时，我们把stack1中的元素逐个弹出并压入stack2。由于先进入队列的元素被压到stack1的栈底，经过弹出和压入之后就处于stack2的栈顶，有可以直接弹出。如果有新元素d插入，我们直接把它压入stack1即可。
  

2.具体步骤：

1. 固定一个栈用于入队操作，另一个栈用于出队操作。
2. 入队列操作，不管stack1和stack2是什么状态，直接向stack1入栈即可。
3. 出队列操作，如果stack2为空，那么把stack1的元素一个一个弹出并压入到stack2中，完成之后，对stack2进行出栈操作，此元素就是最先入队的元素。
4. 出队列操作，如果stack2不为空，那么直接从stack2出栈，此元素是当前所有元素中最先入队的那个。

3.代码：

class Solution
{
public:
    //push操作
    void push(int node) {
        stack1.push(node);
    }
    
    //pop操作
    int pop() {
    	//如果stack2为空
        if(stack2.empty()){    
            while(stack1.size() > 0){
                int data = stack1.top(); //取出栈顶元素
                stack1.pop(); //删除栈顶元素
                stack2.push(data); //压入stack1栈顶元素
            }
        }
        //当stack2不为空时
        int pop_element = stack2.top();
        stack2.pop();
        return pop_element;
    }

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

拓展题目：

用两个队列实现一个栈。

两题异同：
两个栈实现一个队列：使用栈1来入队，栈2来出队，角色是固定了的 。
两个队列实现一个栈：使用非空队列来入栈，使用空队列来出栈，角色会转换。

1.解题思路：

1. 入栈操作：哪个队列非空，就把元素插入哪个队列的队尾，如果两个队列都为空，那么就随便使用一个队列。
2. 出栈操作：把非空队列的元素出队并且按顺序一个一个压入另一个队列，直到剩下一个元素，这个元素就是要出栈的元素。输出该元素即可