思路:
入队:
只将栈s1作为入队。
出队:
将栈s2作为出队,如果栈s1不为空,s2为空,出队时将s1的元素倒入s2中,再pop();
否则s2不为空,出队时直接在栈s2中pop();
#include<iostream> #include<stack> #include<assert.h> using namespace std; template<class T> class Queue { public: Queue() :_size(0) {} void appendTail(T num) { s1.push(num); _size++; } void deleteHead() { assert(_size>0); if(s2.empty()) { while(s1.size()!=1) { s2.push(s1.top()); s1.pop(); } s1.pop(); } else { s2.pop(); } _size--; } T& top() { assert(_size>0); if(s2.empty()) { while(s1.size()!=1) { s2.push(s1.top()); } return s1.top(); } else { return s2.top(); } } private: stack<T> s1; stack<T> s2; int _size; }; int main() { Queue<int> q; q.appendTail(1); q.appendTail(2); q.appendTail(3); q.appendTail(4); q.deleteHead(); q.deleteHead(); q.deleteHead(); q.deleteHead(); q.appendTail(5); cout<<q.top()<<endl; system("pause"); return 0; }