【剑指Offer】用两个栈实现队列

【题目】

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

【思路】

Push操作可以直接把元素存进其中一个栈，我这里是存在stack1栈中。

Pop操作需要结合两个栈。当把元素依次入栈放进栈A，再把栈A的元素依次出栈放进栈B，那么，栈B出栈的顺序就相当于把元素存进队列后的出队顺序。

当第一次pop时，stack2为空，则把stack1的所有元素放进stack2中，再出栈一个元素就行。当stack2有元素时，push操作还是把元素存进stack1栈中，并不会受到影响。当stack1和stack2都有元素时，pop操作只会出栈stack2的元素，直到stack2为空才会再次把stack1的元素放进stack2中。

【代码】

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

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

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