《剑指offer》—5、用两个栈实现队列

题目描述

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

解题思路：

用栈A和栈B来实现一个队列。

入队：直接将元素push入栈A。

出队：判断栈B是否为空，如果为空，将栈A中的所有元素pop，并push进入到栈B，栈B再出栈；如果不为空，栈B直接出栈。

参考代码：

/*************************************************
Copyright:牛客网在线编程《剑指offer》
Author:zhouyuan
Date:2017-02-24
Description:用两个栈实现队列 
**************************************************/

#include <stack>
using namespace std;

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

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

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