1003. 我要通过！(20)--解法2

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于PAT的“答案正确”大派送 —— 只要读入的字符串满足下列条件，系统就输出“答案正确”，否则输出“答案错误”。

得到“答案正确”的条件是：

1. 字符串中必须仅有P, A, T这三种字符，不可以包含其它字符；
2. 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
3. 如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

现在就请你为PAT写一个自动裁判程序，判定哪些字符串是可以获得“ 答案正确 ”的。

输入格式： 每个测试输入包含1个测试用例。第1行给出一个自然数n (<10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过100，且不包含空格。

输出格式：每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出YES，否则输出NO。

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

#include<iostream>
#include<string>
using namespace std;

int main()
{
  /*输入字符串的行数*/
  int n,i,j,j_n[10],a;
  cin >> n;

  /*输入字符串内容*/
  string s[100],spat[100];
  for (i = 0; i < n; i++)
    cin >> s[i];
  /*计算每行字符串有几位*/
  for (i = 0; i < n; i++)
    for (j = 0; s[i][j] != '\0'; j++)
      j_n[i] = j + 1;

/*  for (i = 0; i < n; i++)
    for (j = 0; j < j_n[i]; j++)
      {
      /*判断是否都为PAT，不是则跳出这一行的j循环，进入下一个i循环
      /*如果都是由PAT组成，则赋值给spat这个字符串数组，再进行计算
      if (s[i][j] == 'P' || s[i][j] == 'T' || s[i][j] == 'A')
        spat[i][j] = s[i][j];
      else      
        cout << "NO\n";      
      }
      */
    for (i = 0; i < n; i++)
    {    
      int an = 0, bn = 0, cn = 0;
      if (s[i][0] == 'P')
        {
        if (s[i][1] == 'A'&&s[i][j_n[i]-1] == 'T')
          cout << "YES\n";
        else
          cout << "NO\n";
        }

      if (s[i][0] == 'A')
      {
        /*计算如果是A开头的字符串，A有an个*/
        for (a = 0; s[i][a] == 'A'; a++)
          an = an + 1;

        if (s[i][an] == 'P')
        {
          if (s[i][an + 1] == 'A')
          {
            /*计算P后面是A 的话，有几个连续的A*/
            for (a = an + 1; s[i][a] == 'A'; a++)
              bn = bn + 1;

            if (s[i][an + bn + 1] == 'T')
            {
              if (s[i][an + bn + 2] == 'A')
              {
                /*计算T后面如果是A的话，有几个A*/
                for (a = an + bn + 2; s[i][a] == 'A'; a++)
                  cn = cn + 1;
                if (an == cn&&bn == 1)
                  cout << "YES\n";
                else  
                  if (cn == (an*bn))                
                  cout << "YES\n";
                  else
                  cout << "NO\n";

              }

              else
                cout << "NO\n";
            }
            else
              cout << "NO\n";
          }
          else
            cout << "NO\n";

        }
        else
          cout << "NO\n";

      }
    
      if (s[i][0] != 'P' && s[i][0] != 'A')
        cout << "NO\n";

    }
    
}