3-9 堆栈模拟队列 (20 分)
设已知有两个堆栈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 <bits/stdc++.h>
using namespace std;
int n, m, t;
string s;
stack<int> st1, st2;
int main()
{
cin >> n >> m;
if (n > m)
swap(n, m);
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();
}
else
{
cout << "ERROR:Empty" << endl;
}
}
}
return 0;
}