PAT 甲级 1051 Pop Sequence （25 分)

Note：
1、注意不能提前break或者return，否则可能出现读入数据错乱的情况。
2、我用了一个0来当哨兵，可以省去检测空栈。
3、注意每个新的序列需要重置栈

大致思路：
0、cnt从1开始记录要入栈的数
1、读入pop序列的每一位数num
2、num<top，则返回此序列不可行（flag=0）（不能返回false，要读完序列，故使用flag记录）
3、num>top，则表示还没到这个值，执行如下操作
（1）将num和num之前的数按顺序压入栈s，num在s.top（压入num是为了让cnt到num的下一个值），当size>m+1时，flag=0
（2）pop num
4、num==top，则pop

Code：

#include<iostream>
#include<vector>

using namespace std;

vector<int> s;
int n, m, k;

bool isLegal() {
	int num;
	int cnt = 1;
	s.clear();
	s.push_back(0);
	int flag = 1;
	for (int i = 0; i < n; i++)
	{
		cin >> num;
		if (num > s.back()) {
			while (cnt <= num)
			{
				if (s.size() >= m + 1) flag = 0;
				s.push_back(cnt);
				cnt++;
			}
			s.pop_back();
		}
		else if (num == s.back()) {
			s.pop_back();
		}
		else if (num < s.back()) {
			flag = 0;
		}
	}
	if(flag) return true;
	else return false;
}

int main() {
	cin >> m >> n >> k;
	for (int i = 0; i < k; i++)
	{
		if (isLegal()) {
			cout << "YES" << endl;
		}
		else {
			cout << "NO" << endl;
		}
	}
	return 0;
}