PAT Basic 1003. 我要通过！(20)（C语言实现）

题目

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于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

思路

先理解一下答案正确的条件：

1. 字符串只能包含'P', 'A', 'T'三种字符，简单，有其它字符就是错误的了。
2. xPATx格式的是正确的。这个条件需要正确理解，当时就是在这条上想错了，结果好多次没有通过。两个x是代指相同的字符串。（有些人包括我理解错的原因大概是按照正则表达式的*去理解这里的x了）
3. 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') ; /* need to read the rest of the line, make sure starting from a new line later */    for(int i = 0; i < num; i++)    {        /* three numbers to count 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]++;            else if(c == 'P' && pos == 0)   pos = 1;        /* should have only one P and only before T */            else if(c == 'T' && pos == 1)   pos = 2;        /* should have only one T and only after P */            else                            break;          /* 'wrong' string */        }        if(c == '\n' && pos == 2                            /* correct appearance of P and T, no other chars */         && count[1] && count[2] == count[1] * count[0])    /* relation between three groups of As */            puts("YES");        else            puts("NO");        if(c != '\n') while((c = getchar()) != '\n'); /* still, read the whole line */    }    return 0;}