python 1003 我要通过! (20分)
PAT (Basic Level) Practice(中文) 1003 我要通过! (20分)“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“答案正确”大派送 —— 只要读入的字符串满足下列条件,系统就输出“答案正确”,否则输出“答案错误”。
得到“答案正确”的条件是:
- 字符串中必须仅有
P
、A
、T
这三种字符,不可以包含其它字符; - 任意形如
xPATx
的字符串都可以获得“答案正确”,其中x
或者是空字符串,或者是仅由字母A
组成的字符串; - 如果
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