PAT (Basic Level) Practice （中文）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。

输入样例：

10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO
NO
NO

代码长度限制   16 KB

时间限制 400 ms

内存限制  64 MB

---

分析：

对于字符串输出为 YES的判定要求如下：

一、字符串必须有 P、A、T，并且只包含 P、A、T。

二、分别统计 P左边 A的个数、P与 T之间 A的个数以及 T右边 A的个数，然后对这三个数做数理判断。

第一类字符串：PAT、PAAT等满足条件的字符串为第一类字符串，这一类字符串的特点是第一个数和第三个数为 0，容易判定。

第二类字符串：例如 AAPATAA、AAPAATAAAA，特点是还原后，第二个数为 1。还原指的是循环执行以下操作：当第三个数大于第一个数的时候，第三个数减去第一个数，同时第二个数减去 1。比如前面字符串的 AAPAATAAAA经过一轮还原得到 AAPATAA，是如题所述的形如 xPATx的字符串。而样例中的一个字符串 APAAATAA还原的结果是 APAATA，不满足要求。

代码：

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

int main()
{
    int n, i, cnt[3];
    char s[101], *p, x[101];
    
    scanf("%d\n", &n);
    
    for ( i=n; i; i-- ) {
        
        gets(s);
        
        for ( p=s,cnt[0]=cnt[1]=cnt[2]=0; *p!='\0'; p++ ) {
            if ( *p=='A' ) cnt[0]++; //统计 A的个数
            else if ( *p=='P' ) cnt[1]++; //统计 P的个数
            else if ( *p=='T' ) cnt[2]++; //统计 T的个数
            else break; //遇到非P、A、T的字符跳出循环
        }
        
        if ( *p=='\0' && cnt[1]==1 && cnt[2]==1 && cnt[0] ) { //有且仅有P、A和T，P和T都只能有一个
            
            strcpy( x, strrchr( s, 'T') ); //计算第三个数
            cnt[2] = strlen(x) - 1;
            
            strcpy( x, strchr( s, 'P' ) ); //计算第二个数
            cnt[1] = strlen(x) - cnt[2] - 2;
            
            cnt[0] -= cnt[1] + cnt[2]; //计算第一个数
            
            while ( cnt[0] < cnt[2] ) { //还原
                cnt[2] -= cnt[0];
                cnt[1]--;
            }
            
            if ( ( !cnt[0] && !cnt[2] ) || cnt[1] == 1 ) printf("YES\n");
            else printf("NO\n");
            
        } else printf("NO\n");
    }
    
    return 0;
}

提交结果：