题目
“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 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
解决
先来个演算
发现对于aPbTc, a.length * b.length = c.length; 且b.length > 0, a.length >= 0, c.length >= 0
另外P的位置在T前面
#include <stdio.h>
#include <string.h>
int main(void)
{
int n, len;
char s[10][101];
int cnt[3]; // 记录三个位置A的个数
int P, T; // 记录 P T 个数
int i, j;
scanf("%d", &n);
for (i = 0; i < n; i++)
{
scanf("%s", s[i]);
}
for (i = 0; i < n; i++)
{
len = strlen(s[i]);
cnt[0] = T = P = 0;
cnt[1] = cnt[2] = -1;
for (j = 0; j < len; j++)
{
// 确保字符串无其它字符
if (s[i][j] != 'P' && s[i][j] != 'A' && s[i][j] != 'T')
{
printf("NO\n");
break;
}
if(s[i][j] == 'P')
{
P++;
}else if(s[i][j] == 'T')
{
T++;
}
// P T 个数不能大于1,T不能在P前面(测试点2)
if(P > 1 || T > 1 || T > P)
{
printf("NO\n");
break;
}
cnt[T + P]++;
}
// 上面循环输出了NO
if(j < len)continue;
// 中间没有A 或者 整个字符串没有出现 P 和 T
if(cnt[1] <= 0 || cnt[2] == -1)
{
printf("NO\n");
}else if(cnt[0] * cnt[1] == cnt[2])
{
// 满足条件
printf("YES\n");
}else
{
// 不满足条件
printf("NO\n");
}
}
return 0;
}