【PAT】python 1003 我要通过 (20 point(s))

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

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

字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

这道题应该观察数据形式得到特点：
1.所有正确字符串里只有一个P和T
2.P前和T后对称
3.由条件一和条件二可知正确字符串形如“APATA"、”AAPATAA"，故b为A，a和c相同，在P和T之间再加一个A并在T后加一个a也为正确字符串

总结得到：
遍历字符串若PT之间只有一个A ，则判断P前和T后是否对称。
若之间A的个数大于1，则减少到1，并在T后减去一个P前。
因为只能加“A”所以不要考虑添加字符是否匹配，只需要判断个数是否相等。

           
def judge(s):
    pos_p=0
    pos_t=0
    cnt_p=0
    cnt_t=0
    cnt_mid=0
    for i in range(len(s)):
        if(s[i]=='P'):
            pos_p=i
            cnt_p+=1
        elif s[i]=='T':
            pos_t=i
            cnt_t+=1
        elif s[i]!='A':
            return -1
        if (cnt_t>1 or cnt_p>1):
            return -1
    a=pos_p
    b=pos_t-1-pos_p           //中间A的个数
    c=len(s)-1-pos_t          //T后有多少字符
    if(b<1):
        return -1
    while(b!=1):
        b-=1
        c-=a;
    if(a==c):
        return 1
    else:
        return -1

n=int(input())
for i in range(n):
    s=input()
    if(judge(s)==1):
        print("Yes")
    else:
        print("No")