PAT 乙级 1003

老实说，开始并没有看懂第三个条件什么意思，然后就百度了一下。原来只要‘P’不在第一个就满足：

p前面A的个数乘以PT之间A的个数等于T后面A的个数。

#include<iostream>
using namespace std;

int main()
{
int n;
cin>>n;
while(n--)
{
string s;
cin>>s;
int flag=0;
for(int i=0;i<s.length();i++)
{
if(s[i]!='A'&&s[i]!='P'&&s[i]!='T')
{
cout<<"NO"<<endl;
flag=1;
break;
}
}
int q,p,flag1=0;
if(!flag)
{
for(int i=0;i<s.length();i++)
{
if(s[0]=='P'&&s[s.length()-1]=='T'&&s.length()>2)
{
cout<<"YES"<<endl;
flag1=1;
break;
}

else
{
if(s[i]=='P')
p=i;
if(s[i]=='T')
q=i;
}

}
if(!flag1)
{
if(p*(q-p-1)==s.length()-1-q&&q-p>1)
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}

}

}
return 0;
}