PAT (Basic Level) 1003 我要通过！ (20分)JAVA解法

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		sc.nextLine();
		//List<String> list = new ArrayList<>();
		for (int i = 0; i < N; i++) {
			System.out.println(test(sc.nextLine()));
		}
		
	}

	private static String test(String str) {
		char[] charStr = str.toCharArray();
		int e = 0;
		int m = 0;
		int l = 0;
		int T = 0;
		for (int i = 0; i < charStr.length; i++) {
			if(charStr[i]!='P'&&charStr[i]!='A'&&charStr[i]!='T') {
				return "NO";
			}else if(charStr[i]=='P') {
				e=i;
			}else if(charStr[i]=='T') {
				m=i-e-1;
				T=i;
			}else if(i==charStr.length-1) {
				l=i-T;
			}
		}
		if(m==0) {
			return "NO";
		}
		if(e*m==l) {
			return "YES";
		}
		return "NO";
	}
}