PTA 7-4 出栈序列的合法性

7-4 出栈序列的合法性 添加链接描述
给定一个最大容量为 M 的堆栈，将 N 个数字按 1, 2, 3, …, N 的顺序入栈，允许按任何顺序出栈，则哪些数字序列是不可能得到的？例如给定 M=5、N=7，则我们有可能得到{ 1, 2, 3, 4, 5, 6, 7 }，但不可能得到{ 3, 2, 1, 7, 5, 6, 4 }。

输入格式：
输入第一行给出 3 个不超过 1000 的正整数：M（堆栈最大容量）、N（入栈元素个数）、K（待检查的出栈序列个数）。最后 K 行，每行给出 N 个数字的出栈序列。所有同行数字以空格间隔。

输出格式：
对每一行出栈序列，如果其的确是有可能得到的合法序列，就在一行中输出YES，否则输出NO。

输入样例：
5 7 5
1 2 3 4 5 6 7
3 2 1 7 5 6 4
7 6 5 4 3 2 1
5 6 4 3 7 2 1
1 7 6 5 4 3 2

输出样例：
YES
NO
NO
YES
NO

判断YES或NO的条件：
1.进栈一个出栈一个即1进去就出来 所以顺序就是1-n；
2.看当前栈顶的数与出栈的数是否相等 要是不相等那么肯定有一个数会进栈 判断这个数进栈后有没有超过当前栈的最大容量 超过就肯定是NO 没超过就把这个新的数当成新的栈顶 继续循环进行判断

#include<stdio.h>
int main()
{
	int m,n,k;
	scanf("%d %d %d",&m,&n,&k);
	int a[m];//用于存储栈中的数据
	int b[n];//用于保存出栈的顺序
	int rz,cz;//用于记录入栈与出栈的数
	int i,j;
	int top;//用于记录栈顶 不能超过m 
	int flag;//用来标记是YES还是NO 
	for (i=1;i<=k;i++){ //外循环用于循环查找次数 
		rz=1;cz=0;
		top=0;
		flag=1;
		for (j=0;j<n;j++){ //该循环用于存入入栈的数 
			scanf("%d",&b[j]);
		}
		while(flag==1){
			if (rz==b[cz]){ //这是入栈一个就出栈一个的情况 
		    rz++;
		    cz++; 
		    }//判断出栈的数与栈顶的数是否相等 相等即符合条件 
		    else if(top!=0&&a[top-1]==b[cz]){
		      top--;
		      cz++;
			}
		    else {
			  	if(rz>n){//说明所有的数都已经循环玩了 退出循环 
			  		break;
				}
				a[top]=rz;//因为不满足出栈的条件 所以肯定是在栈中 
				rz++;
				top++;
				if(top>=m){//说明栈中的数已经超过栈的最大容量 那肯定出错了 
					flag=0;
					break;
				}
		    }
		}
		if(flag==0||top!=0){ //判断是否中途退出 
			printf("NO\n");
		}
		else {
			printf("YES\n");
		}
	} 
	return 0;
}