栈是先进后出,队列是先进先出,因此若想用栈保存数据(stack1)获得正确的顺序,就要把栈的顺序反过来。可以把stack1中的数据逐个弹出至另外一个栈(stack2),这样就可以达到正确的队列顺序了
值得注意的是,似乎在出队后需要将stack2的再逐个弹回stack1,实际上不用。在下一次出队时若stack2非空,直接从stack2去弹出;否则,再将stack1的元素全部弹出至stack2,然后再弹出stack2的栈顶元素即可。这样可以避免来回出入栈的时间消耗。
任务:双栈模拟队列
输入:第1行是一个正整数n,表示n个队列运算。接下来每一行一个运算。E x表示元素x入队,D表示出队,F表示输出队首元素
输出:输出F运算的元素,每行一个
#include <iostream>
#include <stack>
using namespace std;
int main()
{
int n;
while( cin >> n )
{
stack<int> stk1, stk2;
char opt;
int x;
while( n-- )
{
cin >> opt;
if( opt == 'E' ){
cin >> x;
stk1.push(x);
}else if( opt == 'F' ){
if( !stk2.empty() )
cout << stk2.top() << endl;
else{
while( !stk1.empty() ){
stk2.push(stk1.top());
stk1.pop();
}
if( !stk2.empty() )
cout << stk2.top() << endl;
else
cerr << "Error! Stack Empty!" << endl;
}
}else if( opt == 'D' ){
if( !stk2.empty() )
stk2.pop();
else{
while( !stk1.empty() ){
stk2.push(stk1.top());
stk1.pop();
}
if( !stk2.empty() )
stk2.pop();
else
cerr << "Error! Stack Empty!" << endl;
}
}
}
cout << "-----END-----" << endl;
}
return 0;
}