【PAT (Basic Level) Practice （中文）（解：Python3.x）】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。

输入样例：

10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO
NO
NO

解析:

条件一 ：任意形如xPATx的字符串都可以获得“答案正确”，其中x或者是空字符串，或者是仅由字符A组成的字符串原字符串中包含'PAT'这个子字符串。且其余字符全是字符'A'

条件二：如果aPbTc是正确的，那么aPbATca也是正确的，其中a,b,c均或者是空字符串，或者是仅由字母A组成的字符串。

可以理解为，有形如'APATA'的字符串，它是正确的，且在中间，和末尾各加一个字符'A'也是“正确答案”，显然这样理解是正确的。那么加多个字符'A'是不是也正确呢？从实验用例来看AAPAATAAAA是正确答案，我们可能会认为是不是在PAT的任意位置加任意个数的字符'A'都是正确的呢？从最后一个实验用例来看显然是错误的。

从其它满分答案来看，有以下约束：

有形如xPyTx的字符串，x是空字符串，或者是仅由字符A组成的字符串，y是仅由字符A组成的字符串，我们以字符P，T进行分段，P前段为a，P T之间为b，T之后为c，则若它是正确答案，有c=a*len(b)

下面我们来看两个实验用例：

1、字符串2AP2AT4A；a段：2A、b段：2A、c段：4A，有 a*len(b)==c。成立！！！

2、字符串1AP3AT2A；a段：1A、b段：3A、c段：2A，有a*len(b)!=c。不成立！！！

综上：这题算法的基本思想就是，以P T为界分为三段来做条件判断。

注：最下面C语言版的是我本人学习顺序问题，本人现在主要的学的放在最前面。

解题:

解法1（使用简单的循环和条件语句实现）：

def test(a):
    x = -1
    y = -1
    for i in range(len(a)):  # 找出P，T的位置
        if (a[i]== 'P'):
            x = i
        if (a[i] == 'T'):
            y = i
    if (x == -1 or y == -1):  # 如果找不到P，T则返回 False
        return False
    if (x > y):  # P在T的后面，返回 False
        return False
    if (y == x + 1):#P，T之间没有字符，返回False
        return False
    if (x != 0):  # 字符串不以P开头
        b = a[0: x]
    else:#字符串以P开头
        b = []
    c = a[x + 1: y]
    if (y != len(a) - 1):  # 字符串不以T结尾
        d = a[y + 1: len(a)]
    else:  # 字符串以T结尾
        d = []
    for i in b:  # 判断各个分段是否是字符A组成
        if (i!='A'):
            return False
    for i in c:
        if (i!='A'):
            return False
    for i in d:
        if (i!='A'):
            return False
    if (b * len(c) == d):  # 条件判断
        return True
    else:
        return False
 

if __name__ == '__main__':
    for i in range(int(input())):
        print('YES' if test(input()) else 'NO')

解法2（正则表达式求解）：

# 正则表达式求解
import re


def yn():
    for i in range(int(input())):  
        rStr = input()  
        if re.match(r'A*PA+TA*',rStr): #在字符串中进行匹配 
            a = re.split(r'[P|T]',rStr)  #以字符P,T进行分段
            if a[0] * len(a[1]) == a[2]:  #条件判断
                print('YES')  # a*len(b)==c
            else:  
                print('NO')  # a*len(b)!=c
        else:  
            print('NO')
            
            
if __name__ == '__main__':
    yn()

解法3（C语言版）：

#include<stdio.h>
#include<string.h>

int main(void)
{
    int i, n, len;
    char str[100];

    scanf("%d", &n);
    while (n--) {
        scanf("%s", str);
        len = strlen(str);
        int pos = 0, count[3] = {0};
        for (i = 0; i < len; i++) {
            if (str[i] == 'A') {
                count[pos]++;
            } else if (str[i] == 'P' && pos == 0) {
                pos = 1;
            } else if (str[i] == 'T' && pos == 1) {
                pos = 2;
            } else {
                break;
            }
        }
        if (pos == 2 && count[1] && count[2] == count[1] * count[0]) {
            puts("YES");
        } else {
            puts("NO");
        }
    }
    
    return 0;
}

如有更好的建议请评论！！！