PAT 1003 我要通过！(20)

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

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

1. 字符串中必须仅有P, A, T这三种字符，不可以包含其它字符；

任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

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

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

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

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

这道题其实有很多坑，不得不吐槽一下PAT的题目了。题目中的条件一和条件二竟然是被包括与包括的关系。所以题目中
P，T的数量只能为1，第二个条件1中的x可以不是同一种字符串。我虽然看不懂题目但是在别人的博客上看，出题人的意
思是条件2中得满足b=n*A, a=m*A, c=k*A，其中k=n*m。

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

int main(){

    int n;
    char str[100];

    scanf("%d",&n);

    while(n--){ //需要验证n个字符串 
        scanf("%s",str); //读入字符串 

        int len = strlen(str);
        int numP = 0, numT = 0, other = 0; //P,T，除PAT以外其他字符的数目
        int posP = -1, posT = -1; //P,T所在下标 
        for(int i = 0; i < len; i++){
            if(str[i] == 'P'){ //如果当前位为P 
                numP++; //P的数量+1 
                posP = i; //记录好P的位置 
            }else if(str[i] == 'T'){ //如果当前位为T 
                numT++; //T的数量+1 
                posT = i; //记录好T的位置 
            }else if(str[i] != 'A'){ //如果当前位不是P,A,T其中之一 
                other++;  //其他字符数量+1 
            }
        }

        if(other > 0 || numP != 1 || numT != 1 || posT-posP<=1 ){ //如果有其他字符，或者PT的数量不是1 ,或者PT之间没有A（这种容易忽略） 
            printf("NO
"); //直接输出NO 
        }else{ //满足条件1 
            int x = posP; //P之前A的个数 
            int y = posT-posP-1; //PT之间A的个数 
            int z = len-posT-1; //T之后A的个数 
            if(x*y == z){ //满足要求 
                printf("YES
");
            }else{ //不满足要求 
                printf("NO
"); 
            }
        } 
    }

    return 0;
}