careercup3.5用两个栈实现一个队列

3.5 Implement a MyQueue class which implements a queue using two stacks.

分析

Queue:first in first out

stack:first in last out

由于队列移除元素移除的是队尾元素，因此用栈s2保持队尾元素。

两个栈

栈s1中:最新的元素在最上面

栈s2中:最旧的元素在最上面

添加元素在栈s1中操作。移除元素时，若栈s2中有元素，则弹出栈顶元素即可。否则把s1中的元素弹出依次压到栈s2中，这样保持最早被压入的元素在s2的栈顶。

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

template<typename T>
class MyQueue{
public:
	MyQueue();
	void add(T elem);
	T top();
	int size();
	T remove();
private:
	stack<T> s1;
	stack<T> s2;

};
template<typename T>
MyQueue<T>::MyQueue(){}
template<typename T>
int MyQueue<T>::size(){
	return s1.size()+s2.size();
}
template<typename T>
void MyQueue<T>::add(T elem){
	s1.push(elem);
}
template<typename T>
T MyQueue<T>::top(){
	if(!s2.empty())
		return s2.top();
	while(!s1.empty()){
		s2.push(s1.top());
		s1.pop();
	}
	return s2.top();

}
template<typename T>
T MyQueue<T>::remove(){
	T elem;
	if(!s2.empty()){
		elem=s2.top();
		s2.pop();
		return elem;
	}
	while(!s1.empty()){
		s2.push(s1.top());
		s1.pop();
	}
	elem=s2.top();
	s2.pop();
	return elem;
}



int main(){

	MyQueue<int> a;
	a.add(5);
	a.add(6);
	a.add(7);
	a.add(8);
	a.add(9);
	a.remove();
	cout<<a.top()<<endl;
	cout<<a.size()<<endl;
	//for test
	system("pause");
	return 0;





}