(PAT乙级)1003.“答案正确”自动判题系统-----我要通过

转载：(PAT乙级)1003.“答案正确”自动判题系统-----我要通过！ - it610.com

 

#include <stdio.h>
#include <string.h>
#define L 101

int main(){
    int n;
    scanf("%d", &n);
    int a,b,c;//a,b,c分别为P之前的A的个数、PT间的A的个数、T之后的A的个数
    char string[L];
    int countp=0;
    int countt=0;
    for (int i = 0; i < n; ++i) {
        countp=0;
        countt=0;
        scanf("%s", string);
        int temp_t ;//temp_t暂时记录数组中字符为T时的下标
        int true = 1;//用于排除不符合题意的情况
        for(int j=0;j<strlen(string);j++){
            if(string[j]=='P')
                countp++;
            if(string[j]=='T')
                countt++;
        }
        if(countt!=1||countp!=1)
            goto end;
        for (int j = 0; j < strlen(string); ++j) {
            if (string[j] == 'P')
                a = j;
            else if (string[j] == 'T')
                temp_t = j;
            else if (string[j] != 'A' )
                true = 0;//若字符串中没有A，则不符合题意
        }
        b = temp_t - a - 1;
        c = strlen(string) - temp_t -1;
        if (b == 0)//若PT之间无A，不符合题意
            true = 0;
        if (c == b * a && true == 1)//由题找规律得：c=ab
            printf("YES\n");
        else
            end:printf("NO\n");
    }
}