天梯赛-L2-032 彩虹瓶 (25分)

#include<iostream>
#include<stack>
#include<algorithm>
#include<vector>
using namespace std;
int main(){
	int N,M,K;
	cin>>N>>M>>K;//彩虹瓶的颜色数量 N、临时货架的容量 M（<N）、发货顺序的数量 K
	vector<int>val;//定义一个vector,用来存放发货的顺序
	for(int i=1;i<=K;i++){
		int ans=1;//用来记录当前正在放的货号
		stack <int> s;//定义一个栈---货架
		for(int i=1;i<=N;i++){//存储输入的发货顺序
			int num;
			cin>>num;
			val.push_back(num);
		}
		for(int j=0;s.size()<=M&&ans<=N;){//循环到货架容量超出所规定的M或者ans达到规定的N
			if(s.size()>0&&s.top()==ans){//先检测货架 先判断栈的大小，再查看栈顶是否存在ans,若相反即栈为空，那么s.top()就会报错
				s.pop();//出栈
				ans++;
			}else{//货架里面没有 
				if(val[j]==ans)//搬来货物查看是否为自己想要的
					ans++;//货号增加
				else//货架上和搬来的货物都不是需要的ans值 
					s.push(val[j]);
				j++;
			} 
		}
		if(ans-1==N) 
			cout<<"YES";
		else 
			cout<<"NO";
		if(i!=K)//判断输出换行
			cout<<endl;
		val.clear();//清空vector 
	}
	
}