B1003 我要通过！（python）

这道题主要是读懂限制条件，理清样例的逻辑关系

• 由题目答案正确的条件可以推断出：
  给出的字符串，仅能包含一个P和T，并且中间不能没有A！
  a、b、c三个位置可以随意插入A，但必须满足：a处A的个数*b处A的个数=c处A的个数

n = eval(input())

for i in range(n):
    ls = input().strip()
    try:
        a = ls[:ls.index('P')]
        b = ls[ls.index('P')+1:ls.index('T')]
        c = ls[ls.index('T')+1:]
    except:
        print('NO')
        continue
    if set(b) == {'A'} and ls == 'P' + b + 'T':
        print('YES')
    elif set(a) == {'A'} and ls == a + 'PAT' + a:
        print('YES')
    elif set(a) == {'A'} and ls == a + 'P' + b + 'T' + 'A'*len(a)*len(b):
        print('YES')
    else:
        print('NO')

优化：

n = eval(input())
for i in range(n):
    ls = input().strip()
    try:
        a = ls[:ls.index('P')]
        b = ls[ls.index('P')+1:ls.index('T')]
        c = ls[ls.index('T')+1:]
        if len(a)*b == c and len(b) != 0:#abc满足的乘数关系（注意a可能为空，不能用a*len(b)），并且b不能为空字符
            print('YES')
        else:
            print('NO')
    except:
        print('NO')