【pat乙級】我要通過！

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

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

字符串中必須僅有 P、 A、 T這三種字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以獲得“答案正確”，其中 x 或者是空字符串，或者是僅由字母 A 組成的字符串；
如果 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

c++代碼：

#include<iostream>
#include<sstream>
using namespace std;
int endpanduan(int k,string a,int l){    //判斷是否T後面還有非A或非空格的字符，有則不正確
    for(int i=k+1;i<l;i++){
        if(a[i]!='A'&&a[i]!=' ')return 0;
    }
    return 1;
}
int panduan(int k,string a,int l){  //判斷複雜情況，判斷A或空格個數a*b?=c(指題目描述3中的a,b,c) 
    int flag=1,sa=k,sb=0,sc=0;   //sa,sb,sc記錄a,b,c字符串字符個數
    if(a[k]!='P')return 0;       //找到開始字符P                                           
    for(int i=k+1;i<l;i++){  //向後遍歷
        if(a[i]==' ')continue;   
        else if(a[i]=='A'){   //找到A
            for(int j=i+1;j<l;j++){     //繼續向後遍歷
                if(a[j]==' '||a[j]=='A')continue;
                else if(a[j]=='T'){   //找到T
                    sb=j-k-1;
                    sc=l-j-1;   //由T的位置劃分3段字符串，得到sa,sc
                    if(sa*sb==sc&&endpanduan(j,a,l)==1)   //判斷sa*sb?=sc，是否T後沒有非法字符
                    return 1;    //條件同時都滿足，返回1，判斷爲正確
                    if(endpanduan(j,a,l)==0)return 0;   //若T後面有非法字符，不在繼續尋找a*b=c，直接判斷爲錯誤
                }
                else {   //遇到的是非A,空格,T字符，非法，判斷爲錯誤
                    return 0;
                }
            }
        }
        else {
            return 0;
        }
    }
    return 0;   //默認判斷爲錯誤(返回0)，因爲只有在循環中找到滿足條件約束的時候才判斷正確
}
int simplepanduan(int k,string a,int l){   //簡單情況下，PAT連着
    if(k+2<l&&a[k]=='P'&&a[k+1]=='A'&&a[k+2]=='T'&&endpanduan(k+2,a,l)==1)return 1; //判斷PAT是否相連，T後是否有非法字符
    return 0;
} 
int main(){
    int n;
    cin>>n;
    getchar();
    for(int i=0;i<n;i++){
        string a;
        getline(cin,a);
        int l=a.length(),flag=0;
        for(int j=0;j<l;j++){
            if(a[j]!='P'&&a[j]!='A'&&a[j]!='T'&&a[j]!=' ')break;  //當前字符非法，判斷爲錯誤
            if(panduan(j,a,l)==1||simplepanduan(j,a,l)==1){  //字符合法，調用判斷函數，判斷當前位置是否滿足條件
                flag=1;break;  //找到則記錄，停止遍歷
            }
        }
        if(flag==0)cout<<"NO"<<endl;  //根據flag判斷是否找到，輸出判斷結果
        else cout<<"YES"<<endl;
    } 
    return 0;
}

注意：
第三條約束條件實際指：
P與T中間每多一個A，T後面就要多加一個P前面的字符串。
即：
…P…T…從左到右用…表示的字符串記爲aa、bb、cc，用a、b、c代表對應的字符個數，應當滿足a*b=c。