1003. 我要通过！(20)

答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于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 <stdio.h>
int Judge(char ch[]){
  int i=0;int tag = 1;int T_time=0,P_time=0;
  int A_time_front=0,A_time_last=0;
  int A_time_mid =0;
  int A_time=0;

  while(ch[i]!='\0'){
    if(ch[i]!='A'&&ch[i]!='P'&&ch[i]!='T'){
      tag=0;
      break;
    }//字符串中必须仅有P,A,T三种字符 
    else if(ch[i]=='P'){
      P_time++;
      if(P_time!=1){
        tag=0;
        break;
      }//记录P的个数 
    }else if(ch[i]=='T'){
        T_time++;
        if(P_time==0||T_time!=1){
            tag = 0;
            break;
        }
    } //T的个数 
    else if(ch[i]=='A'){
        A_time++;
        if(P_time==0){
            A_time_front++;
        } else if(P_time==1&&T_time==0){
            A_time_mid++;
        }else if(P_time==1&&T_time==1){
            A_time_last++;
        }
    }
    i++;
}
    if(P_time==0||T_time==0||A_time==0)
        tag=0;
    else if(A_time_front!=0&&A_time_last!=0){
    if((A_time_last/A_time_front)!=(A_time_mid))
        tag=0;
    }
    return tag;
}
int main(){
  int n,i=0;
  char s[105];
  int result[15];
 scanf("%d",&n);
  for(;i<n;i++){
  scanf("%s",s);
  result[i]=Judge(s);
  }
  for(i=0;i<n;i++){
    if(result[i])
      printf("YES\n");
    else
      printf("NO\n");
  }   
  return 0;
}