L2-032 彩虹瓶

一、题目

二、解题思路

1. 判断输入的数字（搬来的这箱小球）是否符合：若符合，则 cot++ （下一个颜色序号）；若不符合，则入栈（放到临时货架），如果栈溢出（超过临时货架的容量），则标记 flag=false （无法顺利完成任务）。
2. 循环判断栈顶元素（临时货架）是否符合：若符合，则 cot++ （下一个颜色序号），并将栈顶元素移出（从临时货架取下）；否则退出循环。
3. 输出：如果栈溢出（堆积的货物超过容量）或 栈非空（其余颜色的小球无法取下）则表示无法顺利完成任务。

三、代码

#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 ）。