PAT 乙级 1003

1003. 我要通过！(20)

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于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. 正确审题，充分理解正确时候的情况。特别是第二种情况，要将其转换为“中间有多少个A，右边的A的数量就要是左边的多少倍”，这样才足够明确；

2. 循环的终止条件。相信很多同学会在这道题目中用到循环，但这是非常容易出错的。一定要防止数组越界的情况，方法有很多~比如C#中可以利用逻辑符号&&先判断左边、再判断右边的设计，就可以很轻松地避免错误啦~~

using System;

namespace PAT
{
    class Program
    {
        static void Main(string[] args)
        {
            int tryTimes = int.Parse(Console.ReadLine());
            string str;
            for(int i = 0; i < tryTimes; i++)
            {
                str = Console.ReadLine();
                if (ValidStr(str))
                    Console.WriteLine("YES");
                else
                    Console.WriteLine("NO");
            }
        }

        
        static bool ValidStr(string str)
        {
            //先找到P，找到P之前的所有字母必须是A
            int pIndex = -1;    //找到才给这个序号赋值
            int tIndex = -1;    //找到才给这个序号赋值
            int i = 0;
            while(str[i] != 'P' && i < str.Length)
            {
                if (str[i] != 'A')
                    return false;
                i++;
            }
            if (i == str.Length)
                return false;
            else
                pIndex = i;
            //再找T，找到T之前的所有字母必须是A
            i++;
            while(i < str.Length && str[i] != 'T')
            {
                if (str[i] != 'A')
                    return false;
                i++;
            }
            if (i == str.Length)
                return false;
            else
                tIndex = i;

            //接下来再循环，剩下字母必须全是A
            i++;
            while (i < str.Length)
            {
                if (str[i] != 'A')
                    return false;
                i++;
            }

            //接下来验证两种可能的规则
            //第一种可能：左右两边A相等，中间一个
            if (tIndex - pIndex == 2 && pIndex == (str.Length - tIndex - 1))
                return true;
            //第二种可能，aPbTc成立，则aPAbTac成立；即中间有几个A，右边A的数量就要是左边的几倍
            else if (tIndex - pIndex >= 2 && (str.Length - tIndex - 1) == (tIndex - pIndex - 1) * pIndex)
                return true;
            else
                return false;
        }

    }
}