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

Code

#include<bits/stdc++.h>
#define ll long long
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const ll MAXN = 1e12 + 10;
ll n, m, i, j, k, t, w, x, y, z, in, flag, cou, sum, maxn, minn;
string s, s1, s2;
const ll INF = 1e9 + 7;

int main()
{
    FAST_IO;
    cin >> n >> m;

    if ( n > m ) swap ( n, m );

    stack<ll>st1, st2;

    while ( cin >> s && s != "T" )
    {
        if ( s == "A" )
        {
            cin >> t;
            if ( st1.size() < n ) st1.push ( t );
            else if ( st2.empty() )
            {
                while ( !st1.empty() ) st2.push ( st1.top() ), st1.pop();
                st1.push(t);
            }
            else cout << "ERROR:Full" << endl;
        }
        else
        {
            if ( !st2.empty() ) cout << st2.top() << endl, st2.pop();
            else if ( !st1.empty() )
            {
                while ( !st1.empty() ) st2.push ( st1.top() ), st1.pop();

                cout << st2.top() << endl, st2.pop();

                while ( !st2.empty() ) st1.push ( st2.top() ), st2.pop();
            }
            else cout << "ERROR:Empty" << endl;
        }
    }

    return 0;
}