PAT A1051 Pop Sequence 简单思路

题目链接

《算法笔记》上的思路也不算难， 但是并不是我写这道题的第一想法，我们其实可以不用从头比较到尾， 只要我们在某一项遇到不符合条件的，就可以直接输出NO了。我的思路如下：

以 5 6 4 3 7 2 1为例，我要让栈弹出5，首先我要把1 2 3 4 5装进去，然后比较栈顶和弹出去的这个值是否一致，一致，则弹出去，并且下次往栈装数的时候从6开始装，不一致，直接输出NO，读下一个数据。这里5和5是一致的，继续装6，栈顶6和弹出的6是一致的，接下来是4，不用装，比较此时栈顶4和输出4是否一致，一致，继续往下走，直到7个数全输出，最后输出YES。

代码如下：

#include <cstdio>
#include <stack>
using namespace std;
int main(){
	int m, n, k;
	stack<int> s;
	int a[1010]={0};
	scanf("%d %d %d",&m, &n, &k);
	for (int i=0; i<k; i++){
		int j=0,q=1;
		while (!s.empty()){
			s.pop();
		} 
		for (int j=0; j<n; j++){
			scanf("%d",&a[j]);
		}
		for (j=0; j<n; j++){
			while (s.size()<m && q<=a[j]){
				s.push(q++);
			}
			if (a[j]!=s.top()){
				printf("NO\n");
				break;
			}
			s.pop();
		}
		if (j==n) printf("YES\n");
	}
}