原题链接
这道题说实话就是题目有点绕,但其实的意思就是堆栈的一个变式应用。
最核心的代码我写了注释,感兴趣的朋友可以交流一下。
#include <stdio.h>
typedef struct{
int stack[1001];
int top;
}stack;
void init(stack *s){
s->top=0;
}
void push(stack *s,int a){
s->stack[s->top++]=a;
}
void pop(stack *s){
s->top--;
}
int gettop(stack s){
if(s.top!=0){
return s.stack[s.top-1];
}
}
int main(){
int n,m,k,i,j,l,t;
scanf("%d %d %d",&n,&m,&k);
for(l=0;l<k;l++){
int p=1;
stack s;
init(&s);
int x[n],y[n];
for(i=0;i<n;i++){
y[i]=0;
scanf("%d",&x[i]);
}
i=0;
t=1;//设t的初值为1,表示从1号颜色的小球开始装
while(t<=n){//当小球编号小于等于n的时候执行这个循环
//当当前指向的数组中的值不等于t 栈顶的值不等于t 并且t已经入栈过了
//这时工人会发怒
if(x[i]!=t && gettop(s)!=t && y[t]==1){
p=0;
break;
}//当当前指向的数组中的值等于t,i和t都加1
if(x[i]==t){
i++;
t++;
continue;
}//当栈顶的值等于t,退栈一次,t加1
if(gettop(s)==t){
t++;
pop(&s);
continue;
}//当栈顶指针指到m时说明堆栈中已经有m个数了,不能继续入栈
if(s.top==m){
p=0;
break;
}
push(&s,x[i]);//如果都不是上述情况,入栈,i+1
i++;
}
if(p){
printf("YES\n");
}else{
printf("NO\n");
}
}
}