PAT-1003 我要通过 python实现

1.题目

 

本题要好好理解题目中的条件，条件理解错了答案自然就错了。首先统计三种字符的个数，然后找到P和A的下标，由此做分割统计前中后三部分的字符数，最后结合条件做出判断。

2.代码如下 

"""
输入:
10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
XPXXXTAA
APT
APATTAA
输出:
YES
YES
YES
YES
NO
NO
NO
NO
NO
NO
"""
n = int(input())                    # 获取n个测试用例
for i in range(n):                  # 循环n次
    inp = input()                   # 每循环一次获取一个输入的字符串
    P_count = inp.count("P")        # 得到字符串中"P"字符的 个数
    A_count = inp.count("A")        # 得到字符串中"A"字符的 个数
    T_count = inp.count("T")        # 得到字符串中"T"字符的 个数

    P_index = inp.find("P")         # 获取字符串中"P"字符的 下标
    T_index = inp.find("T")         # 获取字符串中"T"字符的 下标
    # 找到前中后"A"的个数(因为是用下标计算的个数，所以不能断定就是字符"A"，但可以先这样理解，后续再通过条件筛选出带有其他字符的错误答案)
    AP = P_index                    # "P"前面字符的个数
    midA = T_index - P_index - 1    # "P"和"T"中间字符的个数
    TA = len(inp) - T_index - 1     # "T"后面字符的个数
    # 1."P"、"A"、"T"的总个数如果等于字符串的长度，那么就相当于不包含其他字符
    # 2.P的个数为1，T的个数为1并且midA大于0，才能保证是形如 xPATx 的(至于要求"P"在"T"前面，下标相减得到midA>0时就已经涵盖在内了)
    # 3.由题，中间每增加一个"A",末尾就要增加一个a字符串，所以AP(a字符串)*midA(中间A的个数)=TA(A个a字符串)
    if P_count + A_count + T_count == len(inp) and P_count == 1 and T_count == 1 and midA > 0 and AP*midA == TA:
        print("YES")
    else:
        print("NO")

 3.提交结果