PAT-我要通过

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“答案正确”大派送 —— 只要读入的字符串满足下列条件，系统就输出“答案正确”，否则输出“答案错误”。
得到“答案正确”的条件是：
字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 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

这个题所指的正确情况由2开始分析：PAT-APATA-AAPATAA-AAAPATAAA
再根据情况三分析：
PAT-PAAT-PAAAT-PAAAAT等等
然后
APATA-APAATAA-APAAATAAA
AAPATAA-AAPAATAAAA-AAPAAATAAAAAA
所以得出结论最开始的a跟c是相等的，再接着是成倍数增加，即T右边的A的个数是最开始a的几倍，那么中间的A就有几个，即左边的A的个数*中间的A的个数=右边的A的个数；
因此只要统计A的个数就可以，但是这里要注意几点，这个字符串有且只能有一个P跟一个T，并且不能含有其他字符，另外字符串中间A的个数必须大于零；

#include <iostream>
#include <vector>
#include <string>
using namespace std;
bool help(string s){
    int p=0,t=0;
    int flag=0;
    int len = s.size();
    for(int i=0; i<len; i++){
        if(s[i]!='P' && s[i]!='A' && s[i]!='T')
            flag++;
        if(s[i] == 'P')
            p++;
        if(s[i] == 'T')
            t++;
    }
    if(p!=1 || t!=1)
        flag++;
    int k,m;//记录P T位置
    for(int i=0; i<len; i++){
        if(s[i]=='P')   k = i;
        if(s[i]=='T')   m = i;
    }
    if(k == m-1)//P T中间没有A
        flag++;
    if((k*(m-k-1)) != (len-1-m))//P前'A'的数量 * P和T之间'A'的数量 应该等于 T后'A'的数量
        flag++;
    if(flag==0)
    {
        return true;
    }
    return false;
}
int main()
{
    int n;
    cin >> n;
    vector<string> s(n);
    for(int i=0; i<n; i++){
        cin >> s[i];
        if( help(s[i]) )
            s[i] = "YES";
        else
            s[i] = "NO";
    }
    for(int i=0; i<n; i++)
        cout << s[i] << endl;
    return 0;
}