【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。