【Java版数据结构】pat乙级1003我要通过（1）

pat乙级1003我要通过！

题目

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“答案正确”大派送 —— 只要读入的字符串满足下列条件，系统就输出“答案正确”，否则输出“答案错误”。

得到“答案正确”的条件是：

字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。
输入格式：

每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (<10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过 100，且不包含空格。
输出格式：

每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出 YES，否则输出 NO。
输入样例：

10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA
结尾无空行

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO
NO
NO
结尾无空行

因为最近在学java,一开始我想用java来写这个算法。分析了一下，字符串有许多方法可以用上。
条件一直接了当
条件二可以用contains(String s)方法判断一个字符串s是否包含在当前字符串中
条件三没看懂题干
最后看了大家的解题思路，最后决定用java来写

**
思路分析
**
条件二： 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；

主要看条件三，如果如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
我们根据条件二已知xPATx是正确的，当a=x,b=A,c=x,且x=A时 APATA是正确的，则APAATAA是正确的
而当x=AA时AAPATAA是正确的，AAPAATAAAA也是正确的
当x=AAA时AAAPATAAA是正确的，AAAPAATAAAAAA也正确
归纳得 当a*b=c时正确
PAT 0乘1=0正确
PAAT 0乘2=0正确
AAPATAA 2乘1=2正确
AAPAATAAAA 2乘2=4正确
但值得注意的地方是 虽然PT满足 0*0=0，本来应该是正确得，但是PT之间至少存在A,故答案错误
…等

代码

package cn.pintia;
import java.util.Scanner;

public class 我要通过1003 {
	public static void main(String[] args){
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        String str[] = new String[n];
        //读入字符串数组
        for(int i=0; i<n; i++){
            str[i] = input.next();
        }
        //循环判断每一个字符串
        for(int i=0; i<n; i++){
            System.out.println(PanDuan(str[i]));
        }
    }
		//判断
    public static String PanDuan(String s){
        int NumA=0,NumP=0,NumT=0;
        int PosP=0,PosT=0;
        for(int i=0; i<s.length(); i++){
            if(s.charAt(i)=='A'){
                NumA++;
            }
            if(s.charAt(i)=='P'){
                NumP++;
                PosP=i;
            }
            if(s.charAt(i)=='T'){
                NumT++;
                PosT=i;
            }
        }
        //长度总和相等，只有一个P，T至少有一个A，T在P后面，当然他们之间至少有一个A,前面A的个数乘以中间A个数等于后面A个数
        if(NumA+NumP+NumT==s.length() && NumA!=0 && NumP==1 && NumT==1 && PosT-PosP>1 && PosP*(PosT-PosP-1)==s.length()-PosT-1 ){
            return "YES";
        }else {
            return "NO";
        }
    }
}