PTA天梯赛L2-032 彩虹瓶（c语言实现）

原题链接
这道题说实话就是题目有点绕，但其实的意思就是堆栈的一个变式应用。
最核心的代码我写了注释，感兴趣的朋友可以交流一下。

#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");
		}
	}
}