1003 我要通过！python实现

import re

num = int(input())
for i in range(num):
    inputStr = input()
    # 在字符串中进行匹配
    if re.match(r'A*PA+TA*', inputStr):
        # 以字符P,T进行分段
        a = re.split(r'[P|T]', inputStr)
        # 条件判断
        # print(a[0], a[1], a[2])
        if a[0] * len(a[1]) == a[2]:
            print('YES')  
        else:  
            print('NO')  
    else:  
        print('NO')