结构与算法 7-22 堆栈模拟队列 (40 行精简)

7-22 堆栈模拟队列 (25 分)

设已知有两个堆栈S1和S2，请用这两个堆栈模拟出一个队列Q。

所谓用堆栈模拟队列，实际上就是通过调用堆栈的下列操作函数:

• int IsFull(Stack S)：判断堆栈S是否已满，返回1或0；
• int IsEmpty (Stack S )：判断堆栈S是否为空，返回1或0；
• void Push(Stack S, ElementType item )：将元素item压入堆栈S；
• ElementType Pop(Stack S )：删除并返回S的栈顶元素。

实现队列的操作，即入队void AddQ(ElementType item)和出队ElementType DeleteQ()。

输入格式:

输入首先给出两个正整数N1和N2，表示堆栈S1和S2的最大容量。随后给出一系列的队列操作：A item表示将item入列（这里假设item为整型数字）；D表示出队操作；T表示输入结束。

输出格式:

对输入中的每个D操作，输出相应出队的数字，或者错误信息ERROR:Empty。如果入队操作无法执行，也需要输出ERROR:Full。每个输出占1行。

输入样例:

3 2
A 1 A 2 A 3 A 4 A 5 D A 6 D A 7 D A 8 D D D D T

输出样例:

ERROR:Full
1
ERROR:Full
2
3
4
7
8
ERROR:Empty

#include <iostream>
#include <stack>
using namespace std;
void moveStack(stack<int>& from, stack<int>& to) {
	while (!from.empty()) {
		to.push(from.top());
		from.pop();
	}
}
int main() {
	stack<int> a,b;
	int n1, n2, e;
	char op;
	cin >> n1 >> n2;
	if (n1 > n2)		//使用小的做输入栈
		swap(n1, n2);
	while (cin >> op) {
		if (op == 'A') {	
			cin >> e;
			if (a.size() < n1) 
				a.push(e);
			else
				cout << "ERROR:Full" << endl;
		}
		else if (op == 'D') {
			if (b.empty() && !a.empty())	//当输出栈为空，输出栈不空的时候转移数据
				moveStack(a, b);
			if (!b.empty()) {
				cout << b.top() << endl;
				b.pop();
			}
			else 
				cout << "ERROR:Empty" << endl;	
		}
		else
			break;
		if (a.size() == n1 && b.empty())	//当输入栈满了，输出栈空了，也需要转移数据
			moveStack(a, b);
	}
	return 0;
}