1003 我要通过!

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“答案正确”大派送 —— 只要读入的字符串满足下列条件,系统就输出“答案正确”,否则输出“答案错误”。

得到“答案正确”的条件是:

字符串中必须仅有 P、 A、 T这三种字符,不可以包含其它字符;
任意形如 xPATx 的字符串都可以获得“答案正确”,其中 x 或者是空字符串,或者是仅由字母 A 组成的字符串;
如果 aPbTc 是正确的,那么 aPbATca 也是正确的,其中 a、 b、 c 均或者是空字符串,或者是仅由字母 A 组成的字符串。
现在就请你为 PAT 写一个自动裁判程序,判定哪些字符串是可以获得“答案正确”的。

输入格式:

每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (≤10),是需要检测的字符串个数。接下来每个字符串占一行,字符串长度不超过 100,且不包含空格。

输出格式:

每个字符串的检测结果占一行,如果该字符串可以获得“答案正确”,则输出 YES,否则输出 NO。

输入样例:

1
2
3
4
5
6
7
8
9
10
11
10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA

输出样例:

1
2
3
4
5
6
7
8
9
10
YES
YES
YES
YES
NO
NO
NO
NO
NO
NO

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#include<stdio.h>
#include<string.h>
int main(){
int i, n;
char str[100];
scanf("%d", &n);
while(n--){
int flag = 1, cl = 0, cm = 0, cr = 0, judge = 0;
scanf("%s", &str);
for(i = 0; i < strlen(str); i++){
if(str[i] == 'P' || str[i] == 'A' || str[i] == 'T'){
if(judge == 0 && str[i] == 'A') cl++;
else if(judge == 0 && str[i] == 'P') judge = 1;
else if(judge == 1 && str[i] == 'A') cm++;
else if(judge == 1 && str[i] == 'T') judge = 2;
else if(judge == 2 && str[i] == 'A') cr++;
else flag = 0;
}else{
flag = 0;
break;
}
}
if(judge != 2 || cl*cm != cr || cm == 0) flag = 0;
if(flag == 1){
printf("YES\n");
}else{
printf("NO\n");
}
}
return 0;
}