题目链接:堆栈模拟队列
题目大意:要求用两个堆栈来模拟一个队列的操作,队列是先进先出,堆栈是先进后出,因此,我们需要两个堆栈来完成这个转换,一个堆栈输出,一个堆栈输入做一个缓冲作用。
核心方法: 找到一个容量最小的堆栈(栈1)来做缓冲堆栈(因为后面需要把这里的数据放入另一个堆栈(栈2),确保转换的时候不会爆栈)
分情况讨论:
1.当c = 'A'时
如果栈1满,栈2不为空, : 输出:ERROR:Full
如果栈1满,栈2为空 :把栈1的所有元素都放入栈2,并且把刚输入的元素存入栈1
栈1没满 : 把刚输入的元素存入栈1
2.当c = 'D'时
如果栈2不为空 : 输出栈顶元素
如果栈2为空,栈1不为空 : 把栈1的所有元素都放入栈2,输出栈顶元素,并删除它
如果栈2为空,栈1为空,输出 : ERROR:Empty
代码如下:
#include<iostream> #include<stack> using namespace std; stack<int> stk1,stk2; int n,m,x; char c; int main(){ cin >> n >> m; getchar(); int t1 = min(n,m); while(1){ cin >> c; if(c == 'T') break; if(c == 'A'){ cin >> x; if(stk1.size() == t1 && stk2.size() > 0){ cout<<"ERROR:Full\n"; } else if(stk1.size() == t1){ while(stk1.size()){ int t = stk1.top(); stk2.push(t); stk1.pop(); } stk1.push(x); } else stk1.push(x); }else if(c == 'D'){ if(stk2.size() > 0){ cout<<stk2.top()<<endl; stk2.pop(); }else if(stk2.size() == 0 && stk1.size() > 0){ while(stk1.size()){ int t = stk1.top(); stk2.push(t); stk1.pop(); } cout<<stk2.top()<<endl; stk2.pop(); }else{ cout<<"ERROR:Empty\n"; } } } return 0; }