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

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

思路：必须知道，stack<> 是先进后出的，queue<>是先进先出的; 所以按照题目要求，例如:我们按次序将1、2、3、4、5分别push进stack中，我们希望得到如图：

这样就达到了题目要求：先进先出。但是现在问题是一个stack无法满足如图需求。我们考虑将1234已经压入stack1，并且满足”先进先出”。现在准备加入5；流程如下图：

//题目描述
//用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。
#include<iostream>
#include<stack>

using namespace std;

stack<int> stack1;
stack<int> stack2;
// |------------
// | 2 3 1 3 5 0
// |------------
void push(int node)
{
    //<1> 将stack1的元素备份到 stack2
    const int stack1_size = stack1.size(); 
    for (int i = 0; i < stack1_size; i++)
    {
        int temp = stack1.top();
        stack1.pop();
        stack2.push(temp);
    }
    stack1.push(node); // push 操作
    // <2> 从stack2恢复元素
    const int stack2_size = stack2.size();
    for (int m = 0; m < stack2_size; m++)
    {
        int temp = stack2.top();
        stack2.pop();//这里顺便把stack2清空了
        stack1.push(temp);
    }
}

int pop() {
    int temp = stack1.top();
    stack1.pop();
    return temp;
}

int main()
{
    push(1);
    push(2);
    push(3);
    push(4);
    push(5);

    cout << pop() << endl; //1
    cout << pop() << endl; //2
    cout << pop() << endl; //3
    cout << pop() << endl; //4
    cout << pop() << endl; //5
    return 1;
}