1003. 我要通过！(20)

复制代码

#include<stdio.h>
#include<string.h>
int main()
{
    char s[1001];
    int i, n, a, b, c;
    char *p;
    scanf("%d", &n);
    for (i = 0; i<n; i++)
    {
        scanf("%s", &s);
        if (strcmp(s, "PAT")==0){ 
            printf("YES\n"); 
            continue;
        }

        a = b = c = 0;
        p = s;
        while (*p == 'A'){ 
            //a的部分
            a++; 
            p++; 
        }
        if (*p == 'P'){
            p++;
            while (*p == 'A'){
                //b的部分
                b++; 
                p++;
            }
            if (*p == 'T'){
                p++;
                while (*p == 'A'){ 
                    //c的部分
                    c++; 
                    p++;
                }
                if (!(*p) && b>0 && c == a*b){ 
                    printf("YES\n"); 
                    continue; 
                }
            }
        }
        printf("NO\n");
    }
    return 0;
}