PTA:7-99 堆栈模拟队列 (25分)--解析

7-99 堆栈模拟队列 (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

思路：
首先要知道堆栈是怎么模拟队列？下面讲讲我个人的理解：
两个栈，一个栈模拟“入列”，每次添加元素时，向这个栈中加入元素；
另一个栈模拟“出列”，每次要取出元素时，从该栈取出；
令s1为入列栈，s2为出列栈
即当执行A命令时，向s1加入元素，如果s1满了，则现先将s1的元素移到s2中，再加入元素。
执行D命令时，从s2中输出元素，如果s2中没有元素了，则将s1中元素移到s2中，然后在输出栈顶元素。
具体代码如下：

#include<bits/stdc++.h>
using namespace std;

int main(){
	stack<int> s1,s2;
	int a, b, item;
	char ch;
	cin >> a >> b;
	if(a>b) swap(a,b);  //s1为小的栈，装着入栈的数，s2装着出栈的数  
	while(1){
		cin >> ch;
		if(ch=='T') break;
		if(ch=='A'){
			cin >> item;
			if(s1.size()==a&&s2.size()!=0) cout << "ERROR:Full"<<endl;
			else{
				if(s1.size()<a) s1.push(item);
				else{  //s1满，s2为空，则将s1全部元素移到s2中 
					while(!s1.empty()){
						s2.push(s1.top());
						s1.pop();
					} 
					s1.push(item);
				}
			}
		}else if(ch=='D'){
			if(s1.empty()&&s2.empty()) cout <<"ERROR:Empty"<<endl;
			else{
				if(!s2.empty()){  //从s2中删除栈顶元素 
					cout << s2.top()<< endl;	
					s2.pop();
				}else{ //如果s2为空，再讲s1中的元素移到s2中，然后在输出删除栈顶元素 
					while(!s1.empty()){
						s2.push(s1.top());
						s1.pop();
					} 
					cout << s2.top() << endl;
					s2.pop();
				}  
			}
		}
	} 
	return 0;
}

欢迎大家批评改正！！！