python 1003 我要通过！ (20分)

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

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

1. 字符串中必须仅有

   P

   、

   A

   、

   T

   这三种字符，不可以包含其它字符；
2. 任意形如

   xPATx

   的字符串都可以获得“答案正确”，其中

   x

   或者是空字符串，或者是仅由字母

   A

   组成的字符串；
3. 如果

   aPbTc

   是正确的，那么

   aPbATca

   也是正确的，其中

   a

   、

   b

   、

   c

   均或者是空字符串，或者是仅由字母

   A

   组成的字符串。

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

输入格式：

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

输出格式：

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

YES

NO

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

实现代码：

def yes(s):
for i in s:
#判断字符串中是否有非'PTA'字符
if i not in 'PAT':
return 'NO'
a=s.find('P')
b=s.find('T')
if a+1==b or a==-1 or b==-1:
#判断字符串中是否有'P','T',且不是'PT'
return 'NO'
# if a+2==b and a+1==len(s)-b:
#如果是'xPATx'形式就需要两侧的A相等(条件2)
#     return 'YES'
if a*(b-a-1)==len(s)-b-1:
#如果是'aPbTc'形式,则a*b=c,可以发现'xPATx'形式是x*1=x,条件2也符合条件3
return 'YES'
return 'NO'
n=int(input())
for i in range(n):
s=input()
print(yes(s))

代码优化：

#正则表达式版:
import re
n=int(input())
for i in range(n):
o='NO'
s=input()
if re.match(r'A*PA+TA*',s):
#判断字符串中是否符合有0个以上的'A','P',有1个以上的'A','T',再有0个以上的'A'
a=re.split(r'[P|T]',s)
#用'P'或'T'分割字符串,python里面可以不加'|',但总觉得难受
o='YES' if a[0]*len(a[1])==a[2] else 'NO'
print(o)

总结：

day3正则表达式真香,就是有时候记不住,所以记不住还打算用的时候,我选择查...

PS: 菜鸟教程的就非常好,W3Cschool也不错,用来当中文api学习最好了
Python 正则表达式: https://www.runoob.com/python/python-reg-expressions.html
正则表达式 - 教程: https://www.runoob.com/regexp/regexp-tutorial.html

• 点赞
• 收藏
• 分享
• 文章举报