1. 程式人生 > >(PAT乙級)我要通過!(Python)

(PAT乙級)我要通過!(Python)

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

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

1. 字串中必須僅有P, A, T這三種字元,不可以包含其它字元;
2. 任意形如 xPATx 的字串都可以獲得“答案正確”,其中 x 或者是空字串,或者是僅由字母 A 組成的字串;
3. 如果 aPbTc 是正確的,那麼 aPbATca 也是正確的,其中 a, b, c 均或者是空字串,或者是僅由字母 A 組成的字串。

現在就請你為PAT寫一個自動裁判程式,判定哪些字串是可以獲得“答案正確”的。

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

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

輸入樣例:
8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
輸出樣例:
YES
YES
YES
YES
NO
NO
NO
NO

利用re模組的match和split來匹配和切割字串;

第一次提交忽略了空格字元,通過elif新增判斷空格字元的情況,第二次提交正確。

import re
n=input('')
for i in range(int(n)):
    s=input('')
    if re.match(r'A*PA+TA*',s):#使用r字首,可忽略轉義問題
        a=re.split(r'[P|T]',s)#用re.split切割。P|T,將P兩端分開,將T兩端分開
        if a[0]*len(a[1])==a[2]:
            print('YES')
        else:
            print('NO')
    elif re.match(r'\s*PA+T\s*',s):
        a=re.split(r'[P|T]',s)
        if a[0]*len(a[1])==a[2]:
            print('YES')
        else:
            print('NO')
    else:
        print('NO')