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;
}