双栈模拟队列

栈是先进后出,队列是先进先出,因此若想用栈保存数据(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;
}

猜你喜欢

转载自blog.csdn.net/Love_Irelia97/article/details/82557910