“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 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
分析:
- 由条件1可以得到,如果读入字符串包含其它字符,则直接输出“答案错误”
- 由条件2和条件3,可以得到(假设用cnt1、cnt2、cnt3分别用来统计’P’之前字符’A’的数量,'P’和’T’之间字符’A’的数量,'T’之后字符’A’的数量 ),则有cnt1*cnt2=cnt3。(从数学递推去找出这个关系)
算法思想
定义一个字符数组进行读入,遍历字符数组:
- 遇到字符‘P’,统计其数目cntP,同时统计‘P’之前‘A’的数量cnt1=i;
- 遇到字符‘T’,统计其数目cntT,同时统计‘P’与‘T’之‘A’的数量cnt2=i-cnt1-1;
- 遇到字符‘A’,统计其数目cnt3;
- 遇到其它字符,直接跳出循环;
- 最后cnt3=cnt3-cnt1-cnt2;当满足cnt2!=0 && cntP1 && cntT1 && cnt3cnt1*cnt2 && istrlen(str)时输出“YES”,否则输出“NO”
代码如下:
#include<stdio.h>
#include<string.h>
#define N 100
int main(){
char str[N];
int n;
int cnt1=0,cnt2=0,cnt3=0,i;
int cntP=0,cntT=0;
scanf("%d",&n);
getchar();//吸收换行字符
while(n--){
//cnt1、cnt2、cnt3分别用来统计'P'之前字符'A'的数量,'P'和'T'之间字符'A'的数量,'T'之后字符'A'的数量
cnt1=0,cnt2=0,cnt3=0;
//cntP、cntT分别用来统计字符'P'和'T'的数量
cntP=0,cntT=0;
gets(str);//读取字符串,gets()以换行结束
for(i=0;i<strlen(str);i++){
if(str[i]=='P'){
cntP++;
cnt1=i;
}else if(str[i]=='T'){
cntT++;
cnt2=i-cnt1-1;
}else if(str[i]=='A'){
cnt3++;
}else{
break;
}
}
cnt3=cnt3-cnt1-cnt2;
//如果'P'和'T'之间字符'A'的数量为0则错误,如果 'P'或者'T'数量超过1则错误
//需要满足cnt3==cnt1*cnt2,并且遍历完毕
if(cnt2!=0 && cntP==1 && cntT==1 && cnt3==cnt1*cnt2 && i==strlen(str))
printf("YES\n");
else
printf("NO\n");
}
return 0;
}