“
答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于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
解题思路:
对于第一个条件只需要遍历字符串判断是否是'P','A','T',不是的话就是答案错误。
对于第二个条件很容易看出符合条件的有"PAT","APATA","AAPATAA",……①规律为T后面A的个数为P前面A的个数的1倍。
对于第三个条件 aPbTc 是正确的,那么 aPbATca 也是正确的。现在根据①知道"PAT","APATA","AAPATAA"是正确的,由此可以知道"PAAT","APAATAA","AAPAATAAAA"……②这个也是正确的。规律为T后面A的个数为P前面A的个数的2倍。
再由②递推可知"PAAAT","APAAATAAA","AAPAAATAAAAAA"……这个也是正确的。规律为T后面A的个数为前面A的个数的3倍。
总结:不难发现,当满足条件为T后面A的个数为P前面A的个数的(PT之间A的个数)倍时,符合题目要求。
#include <cstdio>
#include <cstring>
int main() {
int n;
char temp[105]; //字符数组
scanf("%d", &n);
for(int i = 0; i < n; i++) {
scanf("%s", temp);
int len = strlen(temp);
int pIndex = 0, tIndex = 0; //存放P和T第一次出现的位置
bool flag = true;
//字符串中只要有不是PAT的就标记为false,如果符合要求就记录'P','T'第一次出现的位置
for(int j = 0; j < len; j++) {
if(temp[j] != 'P' && temp[j] != 'A' && temp[j] != 'T') {
flag = false;
break;
}
if(temp[j] == 'P')
pIndex = j;
if(temp[j] == 'T') {
tIndex = j;
break;
}
}
if(flag == false) {
printf("NO\n");
continue;
}
//'P','T'中间的个数小于0的标记为false,因为可能T在P前面,所以会有负数,并判断是不是都为A
int aMidNum = tIndex - pIndex - 1;
if(aMidNum < 0) {
flag = false;
} else {
for(int j = pIndex + 1; j < tIndex; j++) {
if(temp[j] != 'A')
flag = false;
}
}
if(flag == false) {
printf("NO\n");
continue;
}
//统计P前面和T后面A的个数
int aPreNum = 0, aPostNum = 0;
if(pIndex != 0) {
for(int j = 0; j < len;) {
if(temp[j] == 'A' && j < pIndex) {
aPreNum++;
} else if(temp[j] == 'A' && j > tIndex) {
aPostNum++;
} else {
flag = false;
break;
}
if(j == pIndex - 1) j = tIndex + 1;
else j++;
}
}
//如果不符合P前面A的个数乘中间A的个数等于T后面A的个数,标记false
if(aMidNum == 0) {
flag = false;
} else {
if(aPreNum * aMidNum != aPostNum) flag = false;
}
if(flag == true) printf("YES\n");
else printf("NO\n");
}
return 0;
}