这道题主要是读懂限制条件,理清样例的逻辑关系
- 由题目答案正确的条件可以推断出:
给出的字符串,仅能包含一个P和T,并且中间不能没有A!
a、b、c三个位置可以随意插入A,但必须满足:a处A的个数*b处A的个数=c处A的个数
n = eval(input())
for i in range(n):
ls = input().strip()
try:
a = ls[:ls.index('P')]
b = ls[ls.index('P')+1:ls.index('T')]
c = ls[ls.index('T')+1:]
except:
print('NO')
continue
if set(b) == {'A'} and ls == 'P' + b + 'T':
print('YES')
elif set(a) == {'A'} and ls == a + 'PAT' + a:
print('YES')
elif set(a) == {'A'} and ls == a + 'P' + b + 'T' + 'A'*len(a)*len(b):
print('YES')
else:
print('NO')
优化:
n = eval(input())
for i in range(n):
ls = input().strip()
try:
a = ls[:ls.index('P')]
b = ls[ls.index('P')+1:ls.index('T')]
c = ls[ls.index('T')+1:]
if len(a)*b == c and len(b) != 0:#abc满足的乘数关系(注意a可能为空,不能用a*len(b)),并且b不能为空字符
print('YES')
else:
print('NO')
except:
print('NO')