PAT乙级 1003 我要通过 (python)

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

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

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

现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

输入格式：

每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (≤10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过
100，且不包含空格。

输出格式：

每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出 YES，否则输出 NO。

输入样例：

10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO
NO
NO

python代码(以下代码“部分正确”，只能得到17分；最后发现是题目理解错了，居然也能得到17分)（这里犯错误的原因是，仅从输入输出用例去分析规律，以为只要中间满足PAT或者PAAT，前面和后面随便加几个A都行；但实际条件3有更绕的约束条件存在，需要去深入分析才能找到规律）：

# 分析：
# 1,PAT
# 2,PAT,_PAT_,APATA,APAT_,_PATA,（为了方便肉眼看，以下划线代表空字符串）
# 3,b只能是A，a=c= ,则_PAAT__;a=c=A,则APAATAA;a=_,c=A,则_PAATA_;a=A,c=_,则APAAT_A     
# 以及所有的AAPATAA 和 AAPAATAAAA

# 第三步这里分析出错，b并不是只能是A，b是由A组成的字符串，可以是AA,AAA,AAAA......

n = int(input())
s1 = ""
s2 = ""
for i in range(n):
    s3 = []
    s1 = str(input())
    if s1 == 'PAT' or s1 == 'PAAT':
        print('YES')
    else:
        if 'PAT' in s1:
            s2 = s1.replace('PAT','')
            for j in str(s2):
                if j == 'A' or j == "":
                    s3.append('1')
                else:
                    s3.append('0')
            if '0' in s3:
                print('NO')
            else:
                print('YES')
        elif 'PAAT' in s1:
            s2 = s1.replace('PAAT','')
            for j in str(s2):
                if j == 'A' or j == "":
                    s3.append('1')
                else:
                    s3.append('0')
            if '0' in s3:
                print('NO')
            else:
                print('YES')
        else:
            print('NO')

之后参考了这位童鞋的分析，得知是自己题目理解错误（https://blog.csdn.net/qq_41398808/article/details/81051125）。
条件3的分析如下：

有形如’APATA’的字符串，它是正确的，且在中间，和末尾各加一个字符’A’也是“正确答案”，显然这样理解是正确的。那么加多个字符’A’是不是也正确呢？从实验用例来看AAPAATAAAA是正确答案，我们可能会认为是不是在PAT的任意位置加任意个数的字符’A’都是正确的呢？（我就是在这里理解错了），从最后一个实验用例APAAATAA来看显然是错误的。
从其他满分答案来看，有以下约束：
有形如xPyTx的字符串，x是空字符串，或者是仅由字符A组成的字符串，y是仅由字符A组成的字符串，我们以字符P，T进行分段，P前段为a，P T之间为b，T之后为c，则若它是正确答案，有c=a*len(b)
下面我们来看两个实验用例：
①字符串AAPAATAAAA
a段：AA b段：AA c段：AAAA
有 AAAA=AAlen(AA)
AAAA=AA*2，所以是正确答案
②字符串APAAATAA
a段：A b段：AAA c段：AA
有AA=Alen(AAA)
等式不成立，所以是错误答案
理解了以上条件后，就可以很好地解决这道题了，算法的基本思想就是，以P T为界分为三段来做条件判断。

捋清题目规律后，python代码直接用正则匹配，会少写很多代码（答案正确，20分！）：

import re
n = int(input())
for i in range(n):
    s = input()
    if re.match(r'A*PA+TA*',s):        # + 号代表前面的字符必须至少出现一次（1次或多次，所以是PAT,PAAT,PAAAT...）; * 号代表前面的字符可以不出现，也可以出现一次或多次（0次，1次，或多次，满足条件2）
        a = re.split(r'[P|T]',s)       # re.split 多分割符，以P 和 T分隔：P前面为a[0]，P和T中间的为a[1]，T之后的为a[2]
        if a[0]*len(a[1]) == a[2]:     #如果c=a*len(b)
            print('YES')
        else:
            print('NO')
    else:
        print('NO')

————————————————
参考链接：https://blog.csdn.net/qq_41398808/article/details/81051125