一、题目
二、解题思路
- 判断输入的数字(搬来的这箱小球)是否符合:若符合,则 cot++ (下一个颜色序号);若不符合,则入栈(放到临时货架),如果栈溢出(超过临时货架的容量),则标记 flag=false (无法顺利完成任务)。
- 循环判断栈顶元素(临时货架)是否符合:若符合,则 cot++ (下一个颜色序号),并将栈顶元素移出(从临时货架取下);否则退出循环。
- 输出:如果栈溢出(堆积的货物超过容量)或 栈非空(其余颜色的小球无法取下)则表示无法顺利完成任务。
三、代码
#include<iostream>
using namespace std;
#include<stack>
int main()
{int n,m,k;cin>>n>>m>>k;while(k--){stack<int> box;int cot=1;bool flag=true;for(int i=0;i<n;i++){
// 输入的数字(搬来的这箱小球)是否符合 int t;cin>>t;if(cot==t){cot++;}else{box.push(t);if(box.size()>m){flag=false;}}
// 栈顶元素(临时货架)是否符合 while(!box.empty()){if(box.top()==cot){cot++;box.pop();}else{break;}}}
// 栈溢出(堆积的货物超过容量)或 栈非空(其余颜色的小球无法取下)if(flag && box.empty()){cout<<"YES"<<endl;}else{cout<<"NO"<<endl;}}return 0;
}
四、总结
在判断输入的数字是否符合这一步时,若不符合并入栈后,如果栈溢出,标记 flag=false ,需要注意的是此时不要写 break 退出循环,因为数据在持续输入,会导致错误(测试点 1 )。