A1051 Pop Sequence

这题主要自己要想通，不然很难理解
细节在代码注释里面

#include <stdio.h>
#include <algorithm>
using namespace std;

const int maxn = 1010;//最大N
int a[maxn];		//序列
int chk[maxn];		//判断数字状态：未进栈，栈内，出栈
bool Judge(int a[],int capacity,int len)
{
	if(capacity < len){		//说明序列长度大于栈的容量，有可能栈溢出，下面做判断
		for(int i=1;i<=len;i++){
			if(a[i]>capacity && i<=a[i]-capacity)return false;
		}
	}
	fill(chk,chk+len+1,-1); //-1表示没有进栈，0表示在栈内，1表示已经出栈
	int top = 0;			//记录栈顶元素
	for(int i=1;i<=len;i++){
		if(a[i] < top)return false;//比栈顶元素先出栈，不可能
		chk[a[i]] = 1;			//a[i]出栈
		for(int j=a[i]-1;j>=1;j--){
			if(chk[j] == -1)chk[j] = 0;//对于小于a[i]的数字，如果有未进栈的（说明也没有出栈）
		}
		top = 0;	//初始化
		for(int j=a[i]-1;j>=1;j--){	//a[i]出栈后寻找栈顶元素，不一定是a[i]-1的
			if(chk[j] == 0){
				top = j;
				break;
			}
		}
	}
	return true;
}
int main(int argc, char const *argv[])
{
	int m,n,k;
	scanf("%d%d%d",&m,&n,&k);
	for(int i=0;i<k;i++){
		for(int j=1;j<=n;j++)scanf("%d",&a[j]);

		if(Judge(a,m,n))printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}