问题:如何使用两个栈实现一个数列,并写出入队和出队的函数。
栈是后进先出,队列是先进先出。为了模拟先进先出,可以使用两个栈A,B,A栈专门管理入队的元素,B栈专门管理出队的元素。比如要将将1 2 3 4 5入队,则全部压入A中,此时A:1 2 3 4 5。如果要出队,则将A中先全部导入到B中,此时B :5 4 3 2 1 ,这时候再从B栈中出来的就是队列出队的顺序。
#include<iostream>
#include<stack>
using namespace std;
template <typename T>
class Queue{
private:
stack<T>A; //管理入队元素
stack<T>B; //管理出栈元素
public:
void push(T x);
T pop();
};
template <typename T>
void Queue<T>::push(T x){
A.push(x);
}
template <typename T>
T Queue<T>::pop(){
if (B.empty()){ //如果控制出的栈为空
if (A.empty())
return false;
else{
while (!A.empty()){
B.push(A.top());
A.pop();
}
}
}
T x = B.top();
B.pop(); //完成出队列
return x;
}
int main(){
/*测试*/
Queue<int> Q;
int i;
for (i = 0; i < 5; i++)
Q.push(i+1);
for (i = 0; i < 5; i++)
cout << Q.pop() << " ";
cout << endl;
Q.push(1), Q.push(2), Q.push(3);
cout << Q.pop() << endl;
Q.push(4);
cout << Q.pop() << endl;
return 0;
}