解题报告-PAT-Pop Sequence 浙大1051

# include <stdio.h>
# include <stdlib.h>
# include <string.h>
# include <algorithm>
# include <iostream>
# include <stack>
const int MAX_SIZE = 1000+10;
using namespace std;
stack<int >s;
int se[MAX_SIZE];

/*
5 7 2
1 7 6 5 4 3 2
5 6 4 3 7 2 1
*/

/*
问题描述：
给定一个可以容纳M个元素的栈，给定N，以1.2.3.4...N的次序进栈，出栈次序不定，问能否得到
给定的出栈序列。如果能得到，则输出YES，不能得到输出NO。
输入M,N,K。K是测试序列的个数
样例输入:
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
算法思想：
对于给定的序列，5 6 4 3 7 2 1 判断是否是合法的出栈序列，申请指针p，用p依次指向序列每一个数；申请
指针f，记录已经入栈的最大N值，即记录1.2.3.4.5.6.7中入栈的最大N值，因为在入栈的过程中可能会夹杂着出栈，
栈顶元素肯定不是最大的，需要f记录。这样是为了下一次遇到  s.top() < p 需要从f+1开始到p，依次将数压入栈中。
如果 s.top() == p，则栈顶元素出栈，顺序比较下一个数。如果s.top() > p，则序列不合法。 中间如果遇到s.size() > M
的情况，则也要退出。
*/

int main(){
int m,n,k;
cin >> m >> n >> k;
int p,ok,f;
while(k--){
//输入
for(int i=0;i<n;i++){
cin>>se[i];
}
//初始化
ok = 1;
f = 0;
while(!s.empty())
s.pop();

//根据序列依次判断
for(int j=0; j < n; ){
p=se[j];   //当前判断到的数字
if(s.empty() || s.top() < p){//如果栈空就不会执行s.top()
for(int t = f+1; t <= p; t++){
s.push(t);
if(f<s.top())
f=s.top();
}
}
if(s.top() > p){ //s.top() > p一定要放在等于p之前，因为如果栈空，s.top()取不到。
ok=0; break; //但是在中间，就保证了栈不为空。
}
if(s.top() == p){
if(s.size() > m){
ok=0;//出栈之前判断size，入栈也可以判断，但是跳不出两层循环。
break;
}
if(f < s.top())
f = s.top();
s.pop();
j++;
}
}//for
if(!ok) printf("NO\n");
else printf("YES\n");
}

return 0;
}