pta 习题集5-6 堆栈操作合法性

​​S​​和​​X​​分别表示入栈和出栈操作。如果根据一个仅由​​S​​和​​X​​构成的序列，对一个空堆栈进行操作，相应操作均可行（如没有出现删除时栈空）且最后状态也是栈空，则称该序列是合法的堆栈操作序列。请编写程序，输入​​S​​和​​X​​序列，判断该序列是否合法。

输入格式:

输入第一行给出两个正整数N和M，其中N是待测序列的个数，M（$≤50$）是堆栈的最大容量。随后N行，每行中给出一个仅由​​S​​和​​X​​构成的序列。序列保证不为空，且长度不超过100。

输出格式:

​​YES​​如果该序列是合法的堆栈操作序列，或​​NO​​如果不是。

输入样例：

4 10
SSSXXSXXSX
SSSXXSXXS
SSSSSSSSSSXSSXXXXXXXXXXX
SSSXXSXXX

输出样例：

YES
NO
NO
NO

#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include <string>
#include <map>

using namespace std;
int n,m;
char a[105];
int main()
{
  scanf("%d%d",&n,&m);
  for(int i=1;i<=n;i++)
  {
    scanf("%s",a);
    int ans=1;
    int len=strlen(a);
    int t=0;
    for(int i=0;i<len;i++)
    {
      if(a[i]=='S')
      {
        if(t==m)
        {
          ans=0;
          break;
        }
        else
          t++;
      }
      else
      {
        if(t==0)
        {
          ans=0;
          break;
        }
        else
          t--;
      }

    }
    if(t!=0)
      ans=0;
    if(ans)
      printf("YES\n");
    else
      printf("NO\n");
  }
  return 0;
}