3.5 用两个栈实现队列

用两个栈实现队列

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

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

	int pop() {
		if (stack1.empty()) {
			return -1;
		}

		// 确保stack2 为空
		while (!stack2.empty())
		{
			stack2.pop();
		}

		// stack1中的数据全部倒入stack2中
		while (!stack1.empty()) {
			stack2.push(stack1.top());
			stack1.pop();
		}

		// 获取stack2的栈顶元素
		int val = stack2.top();
		stack2.pop();

		// stack2中的数据全部倒回stack1中
		while (!stack2.empty()) {
			stack1.push(stack2.top());
			stack2.pop();
		}

		return val;
	}

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

测试