【PAT_Python解】1003 我要通过！

原题链接：PTA | 程序设计类实验辅助教学平台

Tips：以下Python代码仅个人理解，非最优算法，仅供参考！

def is_pat(s):
    P_start=s.find('P')
    T_start=s.find('T')
    P=s.count('P')
    T=s.count('T')
    A=s.count('A')
    if P !=1 or T!=1 or P+T+A!=len(s):
        return 'NO'
    _0P=s[:P_start]
    _MID=s[P_start+1:T_start]
    _T0=s[T_start+1:]
    if len(_MID) ==0:
        return 'NO'
    if len(_0P) * len(_MID) == len(_T0):
        return 'YES'
    else:
        return 'NO'

n = int(input())
for i in range(n):
    s = input()
    print(is_pat(s))