我的PAT系列文章更新重心已移至Github，欢迎来看PAT题解的小伙伴请到Github Pages浏览最新内容。此处文章目前已更新至与Github Pages同步。欢迎star我的repo。
题目
“ 答案正确 ”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“ 答案正确 ”大派送 ——
只要读入的字符串满足下列条件，系统就输出“ 答案正确 ”，否则输出“ 答案错误 ”。
得到“ 答案正确 ”的条件是：
- 字符串中必须仅有
P、A、T这三种字符，不可以包含其它字符； - 任意形如
xPATx的字符串都可以获得“ 答案正确 ”，其中x或者是空字符串，或者是仅由字母A组成的字符串； - 如果
aPbTc是正确的，那么aPbATca也是正确的，其中a、b、c均或者是空字符串，或者是仅由字母A组成的字符串。
现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“ 答案正确 ”的。
输入格式：
每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 (
)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过
100，且不包含空格。
输出格式：
每个字符串的检测结果占一行，如果该字符串可以获得“ 答案正确 ”，则输出 YES，否则输出 NO。
输入样例：
8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
输出样例：
YES
YES
YES
YES
NO
NO
NO
NO
思路
先理解一下答案正确的条件：
- 字符串只能包含
'P', 'A', 'T'三种字符，简单，有其它字符就是错误的了。 - xPATx格式的是正确的。这个条件需要正确理解，当时就是在这条上想错了，结果好多次没有通过。两个x是代指相同的字符串。（有些人包括我理解错的原因大概是按照正则表达式的
*去理解这里的x了） - aPbTc正确，aPbATca也正确。翻译一下，就是P和T中间每增加一个A，需要将P之前的内容复制到字符串末尾，得到的新字符串就也是正确的。
结合这三点，总结起来就是：
- 只存在
'P', 'A', 'T'三种字符； 'P', 'T'只能出现一次并且按照该顺序先后出现；- P&T之间不能没有A；
- T之后A的数量 = P之前A的数量 × P&T中间A的数量。
代码实现：
使用三个数
count[3]记录P之前，P&T之间，T之后A的数量。-
为了检测
'P', 'T'的出现及次序，使用一个标记变量pos：- 其值在出现P之前为0（使用count[0]记录P之前的A）
- 只有在出现P且其值为0时，将值变为1（使用count[1]记录P&T之间的A）
- 只有在出现T且其值为1时，将其变为2（使用count[2]记录T之后的A）
这样即可保证除此之外出现非'A'字符的情况都是不符合要求的。pos顺便还能作为count[]的索引。
代码
最新代码@github，欢迎交流
#include <stdio.h>
int main()
{
char c;
int num;
scanf("%d", &num);
while(getchar() != '\n'); /* read the rest of the line,
make sure to start from a new line later */
for(int i = 0; i < num; i++)
{
/* pos: indicate the position of current index for count[] */
/* count[3]: number of As; before P, between P & T and after T */
int pos = 0, count[3] = {0, 0, 0};
while((c = getchar()) != '\n')
{
if(c == 'A') count[pos]++; /* count 'A's */
else if(c == 'P' && pos == 0) pos = 1; /* one P before T */
else if(c == 'T' && pos == 1) pos = 2; /* one T after P */
else break; /* 'wrong' string */
}
if(c == '\n' /* 1. no other characters at end */
&& pos == 2 /* 2. appearance of 'P' and 'T' */
&& count[1] /* 3. existance of 'A' */
&& count[2] == count[1] * count[0]) /* 4. relation between numbers */
puts("YES");
else
puts("NO");
/* read the rest of the line */
if(c != '\n')
while(getchar() != '\n');
}
return 0;
}
