题目
“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 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
解题思路
仔细观察答案正确的条件,前面两个容易实现。第三个条件我一开始想到,b必定是一个A,只有这样aPbTc才会正确。后来打脸了( ̄ε(# ̄),这是一个递推的条件,下面我们来模拟一下这个递推过程:
为了方便理解我们把P前面的字符串定义为a,P与T中间的字符串定义为b,T后面的字符串定义为c
1.a=A,b=A,c=A时APATA是正确的,那么APAATAA也是正确的,此时a=A,b=AA,c=AA
2.因为APAATAA是正确的,所以APAAATAAA也是正确的,此时a=A,b=AAA,c=AAA
等等。。。。。。
可以看到a,b,c如果是A组成的字符串他们是有一定的规律。但如果你是按a=A来递推有可能陷入一个狭隘的规则,即b==c答案即正确,但是你使用a=AA再递推一遍就能发现规律了
第三个条件总结出的规律就是:a.length()*b.length()==c.length()
如果a,c是空字符串的话P与T中间有多少个A都是正确的了。
发现规律后程序就容易写出来了,但是还是有一些细节需要注意,我会放在程序的注释中
代码
#include"bits/stdc++.h"
using namespace std;
bool check(string chuan){//检查字符串是不是仅由A组成
for(int i=0;i<chuan.length();i++){
if(chuan[i]!='A')return false;
}
return true;
}
int main(){
string ch;
int n;cin>>n;
while(n--){
cin>>ch;
string a,b,c;
int chang=0,flag=0;
while(ch[chang]!='P'&&chang<ch.length())
a+=ch[chang++];//获取a字符串
while(ch[chang]!='T'&&chang<ch.length())
b+=ch[chang++];//获取b字符串
if(ch[chang]=='T'){//检查是否读到T,非常重要
//比如PA就是没有T的情况,如果没有这一段代码,此时a是空字符串,b=A,c也是空字符串
//符合我们设置的条件,程序会认为这个字符串是正确的,但实际上不符合我们的要求
flag=1;
chang++;
}
while(chang<ch.length())
c+=ch[chang++];//获取c字符串
if(!check(a)||!check(c)){
printf("NO\n");
continue;
}
if(check(b)&&b.length()>0&&flag==1){
if(a.length()*b.length()==c.length()){
printf("YES\n");
continue;
}
else {
printf("NO\n");
continue;
}
}
else printf("NO\n");
}
}