出栈序列的合法性

出栈序列的合法性
给定一个最大容量为 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
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<stack>
using namespace std;
const int p=1009;

int main()
{
	int m,n,k;
	scanf("%d%d%d",&m,&n,&k);
	while(k--)
	{
		int a[p];
		int idx=0,ct=1,flag=0;
		for(int i=0;i<n;++i) scanf("%d",&a[i]);
		stack<int>st;
		while(idx<n)
		{
			if(a[idx]==ct) 
			{
				idx++,ct++;
			}
			else if(st.size()&&a[idx]==st.top()) 
			{
				idx++,st.pop();
			}
		    else {
		    	st.push(ct);
		    	ct++;
		    	if(st.size()>=m)
		    	{
		    		flag=1;
		    		break;
				}
			}
		}
		if(flag==1||st.size()) printf("NO\n");
		else printf("YES\n");
	   
	}
	return 0;
}