【1051】Pop Sequence (25 分)

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>  
#include<map>
#include<vector>
#include<queue> 
#include <stack>
using namespace std;  
const int maxn=1010;
int arr[maxn]; //保存题目给定的出栈序列
stack<int> st;  //定义栈st，用以存放int型元素

int main(){   
  int m,n,T;
  scanf("%d%d%d",&m,&n,&T);
  while(T--){   //循环执行T次
    while(!st.empty()) {  //清空栈
      st.pop();
    }
    for(int i=1;i<=n;i++){
      scanf("%d",&arr[i]);
    }
    int current=1; //指向出栈序列中的带出栈元素
    bool flag=true;
    for(int i=1;i<=n;i++){
      st.push(i); //把i压入栈
      if(st.size()>m){ //如果此时栈中元素个数大于容量m，则序列非法
        flag=false;
        break;
      }
      //栈顶元素与出栈序列当前位置的元素相同时
      while(!st.empty()&&st.top()==arr[current]){
        st.pop();//反复弹栈并令current++
        current++;
      }
    }
    if(st.empty()==true&&flag==true){
      printf("YES\n"); //栈空且flag==true时表明合法
    }else{
      printf("NO\n");
    }
  }
  return 0;   
}