原题链接:PTA | 程序设计类实验辅助教学平台
Tips:以下Python代码仅个人理解,非最优算法,仅供参考!
def is_pat(s):
P_start=s.find('P')
T_start=s.find('T')
P=s.count('P')
T=s.count('T')
A=s.count('A')
if P !=1 or T!=1 or P+T+A!=len(s):
return 'NO'
_0P=s[:P_start]
_MID=s[P_start+1:T_start]
_T0=s[T_start+1:]
if len(_MID) ==0:
return 'NO'
if len(_0P) * len(_MID) == len(_T0):
return 'YES'
else:
return 'NO'
n = int(input())
for i in range(n):
s = input()
print(is_pat(s))