Q:Implement a MyQueue class which implements a queue using two stacks.
A:一个栈sin用于进,一个栈sout用于出。
push:直接push进sin即可
pop: 如果sout不空,直接pop,否则将sin中的元素都压入sout之后再sout.pop
front: 同pop,只不过不pop,而是top
back:同front, 如果sin不为空,直接sin.top(),否则将sout元素都压入sin, 再sin.pop
#include <iostream>
#include <stack>
using namespace std;
template <typename T>
class MyQueue{
private:
stack<T> sin, sout;
public:
MyQueue() {
}
~MyQueue() {
}
void push(T val){
sin.push(val);
}
void pop(){
move(sin, sout);
sout.pop();
}
T front(){
move(sin, sout);
return sout.top();
}
T back(){
move(sout, sin);
return sin.top();
}
int size(){
return sin.size()+sout.size();
}
bool empty(){
return sin.empty()&&sout.empty();
}
void move(stack<T> &src, stack<T> &dst){
while (dst.empty()) {
while (!src.empty()) {
dst.push(src.top());
src.pop();
}
}
}
};
int main() {
MyQueue<int> q;
MyQueue<int> q1;
for(int i=0; i<10; ++i){
q.push(i);
q1.push(i);
}
cout<<q.front()<<" "<<q.back()<<endl;
cout<<q1.front()<<" "<<q1.back()<<endl;
cout<<endl;
q.pop();
q1.pop();
q.push(10);
q1.push(10);
cout<<q.front()<<" "<<q.back()<<endl;
cout<<q1.front()<<" "<<q1.back()<<endl;
cout<<endl;
cout<<q.size()<<" "<<q.empty()<<endl;
cout<<q1.size()<<" "<<q1.empty()<<endl;
return 0;
}