1003. 我要通過！(20)

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

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

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

輸入格式： 每個測試輸入包含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 } 

PAT 1003.我要通過！