1003 我要通过！python

1003 我要通过！ (20 分)

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

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

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

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

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

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

PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO
分析：
1.字符串中必须仅有P, A, T这三种字符，不可以包含其它字符；\

2.任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；\

3.如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

分析第1个条件，可定义3个标记P,A,T是否存在于字符串中的标识变量，遍历字符串，如果有不为P,A,T的字符或者字符串由P,A,T中的一个或两个字符构成，则不符合要求。

分析第2个条件，x如果不为空，则一定是由A构成的串，且P前的x=T后的x

分析第3个条件，如果aPbTc正确，那么aPbTc一定由xPATx构成或由xPATx根据条件3推导而来，而aPbTc最初的状态一定是xPATx，即b=A,a=c=空串或n个A。现在P之前有一个a，T之后有一个a，PT之间有一个A，如果P和T之间增加一个A，那么c后面就增加一个a,按照规律，假设PT之间存在m个A，T之后存在n个a，因为m和n都从1开始且它们每次的增量相等，则：

m=n

假设a部分存在k个A，则

km=nk

即P之前A的个数k*PT之间A的个数m=T之后A的个数nk，不妨设为式1。

不难发现，第3个条件建立在字符串仅由P,A,T构成且只有一个P和T，P和T之间只有A的前提下，故在根据第1个条件作相应操作后，还需“过滤”多余的P和T且保证P和T之间只有A。方法一般可为遍历字符串，找到第一个P和最后一个T的位置，在这两个字符之间判定是否有不为A的字符。最后根据串长，两个位置可得到式1所需参数，判等即可。
以上分析引用下面链接：

https://blog.csdn.net/qq_37729102/article/details/81589739

n=input()
n=int(n)
a=[]
for i in range(n):
    t=input()
    a.append(t)
for k in a:
    if ('P'  in k)&('A'in k)&('T'in k):
        x=k.count('P')
        y=k.count('A')
        z=k.count('T')
        if len(k)==x+y+z:
            if x==1 & z==1:
                if k.index('P')<k.index('T'):
                    temp1=k[:k.index('P')]
                    temp2=k[k.index('P'):k.index('T')]
                    temp3=k[k.index('T'):]
                    x1=temp1.count('A')
                    y1=temp2.count('A')
                    z1=temp3.count('A')
                    if z1==x1*y1:
                        print('YES')
                    else:
                        print('NO')
                else:
                    print('NO')
            else:
                 print('NO')
        else:
            print('NO')
    else:
        print('NO')
  

---

通过以上代码学习到的知识点：

#查找'P'是否存在字符串K中
'P'  in k
#对字符串中含有P进行统计
k.count('P')
#对字符串进行切片,从a-b（包括k[a],k[b]在内）
temp=k[a:b]
#找到T的索引
k.index('T')
#求字符串的长度
len(temp)