1003. 我要通過！(20) python篇__python

“答案正確”是自動判題系統給出的最令人歡喜的回複。本題屬於PAT的“答案正確”大派送 —— 只要讀入的字串滿足下列條件，系統就輸出“答案正確”，否則輸出“答案錯誤”。

得到“答案正確”的條件是：

1. 字串中必須僅有P, A, T這三種字元，不可以包含其它字元；2. 任意形如 xPATx 的字串都可以獲得“答案正確”，其中 x 或者是Null 字元串，或者是僅由字母 A 組成的字串；3. 如果 aPbTc 是正確的，那麼 aPbATca 也是正確的，其中 a, b, c 均或者是Null 字元串，或者是僅由字母 A 組成的字串。現在就請你為PAT寫一個自動裁判程式，判定哪些字串是可以獲得“ 答案正確”的。

輸入格式：每個測試輸入包含1個測試案例。第1行給出一個自然數n (<10)，是需要檢測的字串個數。接下來每個字串佔一行，字串長度不超過100，且不包含空格。

輸出格式：每個字串的檢測結果佔一行，如果該字串可以獲得“答案正確”，則輸出YES，否則輸出NO。 輸入範例：

8PATPAATAAPATAAAAPAATAAAAxPATxPTWhateverAPAAATAA

輸出範例：

YESYESYESYESNONONONO

import ren=input()for i in range(int(n)):    s=input()    if re.match(r'A*PA+TA*',s):        a=re.split(r'[P|T]',s)        if a[0]*len(a[1])==a[2]:            print('YES')        else:            print('NO')    else:        print('NO')