PAT 1051. Pop Sequence (25)

模拟进栈出栈问题。注意栈有大小。

#include<iostream>
#include<cstdio>
#include<stack>
using namespace std;


int main()
{
    int M,n,k;
    while(scanf("%d%d%d",&M,&n,&k) != EOF)
    {
        for(int i = 1; i <= k; i++)
        {
          stack<int> s;
          int target[1004];
          int A = 1, B = 1,ct = 0;
          for(int j = 1; j <= n; j++)
              scanf("%d",&target[j]);
          bool Ok = true;
          while(B <= n)
          {
              if(ct < M && A == target[B])//相等就相当于这个元素进栈立刻又出栈
              {
                  A++;
                  B++;
              }else if(!s.empty() && s.top() == target[B])//栈顶元素弹出
              {
                  s.pop();
                  B++;
                  ct--;
              }else if(A <= n && ct < M)//进栈
              {
                     s.push(A++);
                     ct++;
              }
              else
              {
                  Ok = false;
                  break;
              }
          }
          printf("%s\n",Ok? "YES":"NO");
        }
    }
    return 0;
}