PAT 1003.我要通过!
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 }