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

实现一

思路

    s1是入栈的，s2是出栈的。

• 入队列，直接压到s1是就行了
• 出队列，先把s1中的元素全部出栈压入到s2中，弹出s2中的栈顶元素；再把s2的所有元素全部压回s1中 

代码如下：

#include<iostream>
#include<stack>
using namespace std;

class NewQueue
{
private:
	stack<int> s1,s2;

public:
	//入队
	void enter(int value)
	{
		s1.push(value);
	} 
	
	//出队
	int out() 
	{
		if(s2.empty()) //若栈s2为空 
		{
			while(!s1.empty()) //栈s1非空时
			{
				int value=s1.top();
				s2.push(value);
				s1.pop();
			} 
		}
		

		int topValue=s2.top();
		s2.pop();
		return topValue;
		
	}
	
	bool QueueEmpty()
	{
		return (s1.empty() && s2.empty());
	}
}; 

int main()
{
	NewQueue queue;
	int N,num;
	cin>>N;
	
	for(int i=0;i<N;i++)
	{
		cin>>num;
		queue.enter(num);
	}
	
	while(!queue.QueueEmpty())
	{
		cout<<queue.out()<<" ";
	}
	
	return 0;  
}