（刷题记录）PAT（乙级 ）1003 我要通过！

问题描述

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 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这三个字符，出现其他字符都算错误。

其次，第二个条件的xPATx也很好懂，x要么为空字符串，要么为只有字母A的字符串。也就是说，只能出现一个PAT字符串，其他位置若不为空则仅能为A，并且，P前的字符串长度要等于T后的字符串长度。

然后，第三个条件才是关键，若aPbTc正确，则aPbATca也正确。这个条件是第二个条件的延伸，因为在此条件下判断字符串aPbATca是否正确，必定要判断其aPbTc形式是否正确，这样不停往前追溯，最终会回到xPATx（或写作aPATa）形式。因为每次迭代P、T中间差一个A，T后差一个a，即，b的长度为迭代的次数n，c的长度为n * a，这里我们设B=len(bA)，C=len(ca)，可得n=B-1，C=(n+1) * a= B * a。
这样我们就有了判断第三个条件正误的表达式 C == B * a，其他就是小事儿了。

出现过的问题

主要问题还是第三个条件的判定上，当然，新手的我出现了其他问题。

读取样例字符串时，原本使用的是gets（），结果PAT的OJ没通过（现在想来，应该是数组大小设置问题，题中说长度不超过100，我居然真的设成了100.。。。忘了末尾的‘\0’了）。

当然，当时我还没意识到这个问题，以为是gets溢出了，于是改用了fgets，编译通过但结果依旧不对，查询后发现fgets输入的末尾会有个‘\n’，而我循环的终止条件里用的‘\0’，修改后样例也正常通过了，信心满满上传PAT的OJ，却还是有一个测试没过。看着fgets的第二个参数，突然发现是数组大小有问题，大小改为101也不对，然后我查了一下fgets（），发现末尾是先写入一个’\n’，再写入一个’\0’，嗯，那就是102了，上PAT的OJ测试，没问题，解决了！

通过后，我又继续使用了scanf（“%s” , str );来读入字符串，数组大小设为101也没问题能通过。随后，在提交系统中我发现了不同，fgets输入时，测试耗时为2s；scanf输入时，测试耗时为3s（知识+1）.

代码

#include <stdio.h>
#include <math.h>
#include <string.h>
#include <stdlib.h>

#define M 10
#define N 102

int Jud(char *str)
{
    int ct[3]={0}; //记录P、T、A各字母出现次数
    int site[2]={0}; //记录P、T各字母出现位置
    int i;
    for(i=0;str[i]!='\n';i++) //检查是否有P、A、T以外的字母存在
    {
        if(str[i]=='P' )
        {
            ct[0]++;
            site[0]=i;
        }
        else if(str[i]=='T')
        {
            ct[1]++;
            site[1]=i;
        }
        else if(str[i]=='A')
        {
            ct[2]++;
        }
        else return 0;
    }

    if(ct[0]!=1 || ct[1]!=1 || ct[2]==0 || site[0]>site[1])
    {
        return 0; //若P、T出现次数大于1，或没有A，或P在T后面则错误
    }

    int a,B,C;
    a=site[0];
    B=site[1]-site[0]-1;
    C=i-1-site[1];

    if (C==B*a)
        return 1;
    else
        return 0;
}

int main(void)
{
    int n;
    char str[N]={0};
    int outp[M]; //记录每个字符串的正误

    scanf("%d",&n); //读入字符串个数
    getchar();

    for (int i=0; i<n; i++) //输入字符串并记录判定结果
    {

        fgets(str,N,stdin);
        outp[i]=Jud(str);

    }

    for (int i=0; i<n; i++) //输出判定结果
    {

        if(outp[i])
            printf("YES\n");
        else
            printf("NO\n");

    }

    return 0;
}