PAT (Basic Level) Practice （中文）1003 我要通过！python

出处：https://pintia.cn/problem-sets/994805260223102976/problems/994805323154440192
参考：https://blog.csdn.net/nanhuaibeian/article/details/98741857?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522161900721816780264041480%2522%252C%2522scm%2522%253A%252220140713.130102334…%2522%257D&request_id=161900721816780264041480&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2_allsobaiduend~default-1-98741857.pc_search_result_before_js&utm_term=1003+我要通过！+python

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

1.字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
2.任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
3.如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

这个题目写的有点谜语人，参考其他人的答案，对3的解释对于一个输入，T后边的A的数量必须是P前A的数量与P和T之间A的数量的乘积。

import re   #导入正则表达式
n = int(input())    #n为输入字符串的个数
for i in range(n):  #对每一个输入的字符串测试是否符合条件
    temp = input()
    if re.match(r'A*PA+TA*',temp):  #字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符,且其他部分都用A填充
                                    #* :(>=0)表示其前边的字符需要出现0次或者多次
                                    #+ :(>=1)其前边的字符需要出现1次或者多次
        a = re.split(r'[P|T]',temp)      #a表示用P或T将字符串分割开，用来测试3
                                    #| :表示或
                                    #[]表示要匹配的字符只能取材于[]里面
        if len(a[0]) * len(a[1]) == len(a[2]):  #也可写成 if len(a[0]) * a[1] == a[2]:
            print('YES')
        else:
            print('NO')
    else:
        print('NO')