【PAT B1003】我要通过!（C语言）

#include <stdio.h>

int main() {
    int p, a, t;
    p = 'P';
    a = 'A';
    t = 'T';
    int flag[10] = {0};//判断是否有PAT其他的字母
    int count[10][3] = {0};//判断是否PAT都有
    int loca[10]= { 0 };//判断A的位置
    int sum[10][3] = { 0 };//计数各位置A的个数
    int in[10][102] ;
    char out[10][5] = {'\0'};
    int n,i=0,j=0;
    scanf("%d", &n);
    getchar();
    while (i < n) {
        j = 0;
        while (in[i][j] != 10) {
            in[i][j] = getchar() ;
            if (in[i][j] == 10) 
                break;
            if (in[i][j] == p) {
                if (loca[i] == 0) {
                    loca[i] = 1;
                }
                else {
                    flag[i] = 1;
                }
                count[i][0]++;
            }
            else if (in[i][j] == a) {
                sum[i][loca[i]]++;
                count[i][1]++;
            }
            else if (in[i][j] == t) {
                if (loca[i] == 1) {
                    loca[i] = 2;
                }
                else {
                    flag[i] = 1;
                }
                count[i][2]++;
            }
            else {
                flag[i] = 1;
            }

            j++;
        } 
            if (flag[i] != 1 && count[i][0] == 1 && sum[i][1] != 0 && count[i][2] == 1 && sum[i][0] * sum[i][1] == sum[i][2]) {
                out[i][0] = 'Y';
                out[i][1] = 'E';
                out[i][2] = 'S';
                out[i][3] = '\n';
            }
            else {
                out[i][0] = 'N';
                out[i][1] = 'O';
                out[i][2] = '\n';
            }
            i++;
    }


    for (i = 0; i < n; i++) {
        printf("%s", out[i]);
    }

    return 0;
}