1003. 我要通过！(20)

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于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

 1 #include<iostream>
 2 #include<algorithm>
 3 using namespace std;
 4 int flag=1;
 5 int IsABC(string s){
 6     if(s=="") return 1;
 7     for(auto b=s.begin();b<s.end();b++)
 8     if(*b!=‘A‘&&(*b)!=‘\0‘) return 0;
 9     return 1;
10 }
11 int main(){
12     string str;
13     int N;
14     cin>>N;
15     for(int i=0;i<N;i++){
16         auto j=str.begin();
17     cin>>str;
18     do{
19     flag=1;
20     j=str.begin();string a,b,c;
21     for(auto i=str.begin();i<str.end();i++){
22     if(*i==‘P‘){
23     copy(j,i,inserter(a,a.begin()));
24     j=i+1;
25     if(!IsABC(a)) {flag=0; break;}
26     }
27     if(*i==‘T‘){
28     copy(j,i-1,inserter(b,b.begin()));
29     copy(i+1,str.end()-a.length(),inserter(c,c.begin()));
30     if(!IsABC(b)) {flag=0; break;}
31     if(!IsABC(c)) {flag=0; break;}
32      }
33     }
34     if(flag==0)  break;
35     if(b[0]!=‘A‘&&c[0]!=‘A‘){
36      if(str.length()!=a.length()*2+3) flag=0;
37      for(int i=0;i<a.length();i++)
38      if(str[i]!=‘A‘) flag=0;
39      for(int i=a.length()+3;i<str.length()-1;i++)
40      if(str[i]!=‘A‘)  flag=0;
41      if(str[a.length()]!=‘P‘) flag=0;
42      if(str[a.length()+1]!=‘A‘) flag=0;
43      if(str[a.length()+2]!=‘T‘) flag=0;
44      break;
45     }
46     str=a+"P"+b+"T"+c;
47     }while(1);
48     if(flag==1) cout<<"YES"<<endl;
49     else cout<<"NO"<<endl;
50     }
51     return 0;
52 }