PAT乙级1003.我要通过

Question

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        int i=0,n,j;
        int YN=1;  
        int p1=0,t1=0;
        int p2=0,t2=0;
        Scanner input =new Scanner(System.in);
        n=input.nextInt();
        input.nextLine();
        for(j=0;j<n;j++) { 
            String arr=input.nextLine();
            p2=0;t2=0;
            YN=0;
            int flag1=0,flag=0;
            char[] aa=arr.toCharArray();
            for(i=0;i<aa.length;i++){
                if(aa[i]=='P'){
                    p1=i;
                    p2++;
                }
                if(aa[i]=='T'){
                    t1=i;
                    t2++;
                }
                if(aa[i]=='A'||aa[i]=='P'||aa[i]=='T') flag1++;
            }
            if(flag1==aa.length) flag++;
            if(p1*(t1-p1-1)==(aa.length-1-t1)) flag++;
            if(p2==1&&t2==1) flag++;
            if((t1-p1)>=2) flag++;
            if(flag==4) System.out.println("YES");
            else System.out.println("NO");
        }
    }
}

为什么全错？不至于吧？我为什么错了？

Answer 1

 #include <cstdio>
#include <cstring>
int main() {
    int n;
    char str[110];
    scanf("%d", &n);
    while (n--) {
        int flag = 0, pcount = 0, tcount = 0, pindex = -1, tindex = -1;
        scanf("%s", str);
        int len = strlen(str);
        for (int i = 0; i < len; ++i) {
            if (str[i] != 'P' && str[i] != 'A' && str[i] != 'T') {
                flag = 1;
                break;
            }
            if (str[i] == 'P') {
                pindex = i;
                pcount++;
                if (pcount > 1) {flag = 1; break;}  //若P的数量大于一个则必定为NO，T下同。
            }
            if (str[i] == 'T') {
                tindex = i;
                tcount++;
                if (tcount > 1) {flag = 1; break;}
            }

        }
        if ( (len - tindex - 1) - (pindex) * (tindex - pindex - 2) != (pindex) || (tindex - pindex - 2 < 0)) {flag = 1;}
        if (flag)printf("NO\n");
        else printf("YES\n");
    }
    return 0;
}