【PAT乙级】1003 我要通过！（JAVA实现）

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

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

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

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

输入格式：

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

输出格式：

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

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES YES YES YES NO NO NO NO

解题思路：

题目中对于输入字符串的限定有三条：

1. 字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；

可以为其设定一个方法进行检测，观察其内部是否包含除P、A、T以外的字符。

2. 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；

此规则用以保证：
1、P、T中间一定有A隔开
2、P、T左右两侧一定对称（都为空，或者都为相同长度的A）

3. 如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A组成的字符串。

此规则说明，字符串尾可以扩充，扩充的规则为：
1、每次增加P字符左侧的字符段（空字符或由A组成的字符串）
2、每增加一次，P、T中间多增加一个A（由于左右两侧都可以是空字符串，所以P、T中间可以有任意多的A）

在此思路下，代码实现如下：

import java.util.Scanner;

public class Main
	{

		public static void main(String[] args)
		{
			// TODO Auto-generated method stub
			Scanner in = new Scanner(System.in);
			//完成数据的读取
			int num = in.nextInt();
			String[] testArray = new String[10];
			for (int i = 0; i < num; i++)
			{
				testArray[i] = in.next();
			}
			in.close();
			//使用isCorrect方法循环检测是否符合规则
			for (int i = 0; i < num; i++)
			{
				if (isCorrect(testArray[i]))
				{
					System.out.println("YES");
				} else
				{
					System.out.println("NO");
				}
			}
		}
		//规则分别实现,使用该方法进行集成
		static boolean isCorrect(String tmp)
		{
			char[] charArray = tmp.toCharArray();
			if (test1(charArray) && test2(charArray))
			{
				return true;
			} else
			{
				return false;
			}
		}

		// 实现规则1
		static boolean test1(char[] charArray)
		{
			for (int i = 0; i < charArray.length; i++)
			{
				if (charArray[i] != 'A' && charArray[i] != 'P' && charArray[i] != 'T')
				{
					return false;
				}
			}
			return true;
		}

		// 实现规则2、3
		static boolean test2(char[] charArray)
		{
			int posP = search(charArray, 'P');
			int posT = search(charArray, 'T');
			//若字符串中无P或T
			if(posP == -1||posT == -1)
			{
				return false;
			}
			//若两侧为空字符串
			if (posP == 0)
			{
				//需保证P、T之间A的个数至少为1个，且T为字符串尾
				if (posT - posP > 1&&posT == charArray.length - 1)
				{
					return true;
				}
			} else
			{
				//求得尾部的A的个数是首部的A的个数的几倍
				int count = (charArray.length - 1 - posT) / posP;
				//P、T之间A的个数应满足倍数，且尾部A的个数是首部A的个数的整数倍
				if (posT - posP == count + 1 && (charArray.length - 1 - posT) % posP == 0)
				{
					return true;
				}
			}
			return false;
		}
		//简单的线性查找方法
		static int search(char[] charArray, char key)
		{
			for (int i = 0; i < charArray.length; i++)
			{
				if (charArray[i] == key)
				{
					return i;
				}
			}
			return -1;
		}
	}