PAT 1003 我要通过！ python

最新推荐文章于 2024-09-12 07:00:00 发布

Raaay233

最新推荐文章于 2024-09-12 07:00:00 发布

阅读量746

点赞数

分类专栏： PAT Python 文章标签： PAT 1003

本文链接：https://blog.csdn.net/qq_42800234/article/details/81537326

版权

PAT 同时被 2 个专栏收录

95 篇文章 5 订阅

订阅专栏

Python

95 篇文章 1 订阅

订阅专栏

1003 我要通过！(20)（20 分）

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

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

字符串中必须仅有P, A, T这三种字符，不可以包含其它字符；\
1. 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；\
2. 如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
  现在就请你为PAT写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

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

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

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

作者: CHEN, Yue
单位: PAT联盟
时间限制: 400ms
内存限制: 64MB
代码长度限制: 16KB

问题分析：
1. 由条件一可知，PAT、APATA、AAPATAA等为正确答案
2. 条件二需要在条件一的基础上，也就是说b只能为空字符串或者一个A，并且a与c的数量应一样

代码：

import re

nums = input()  # 接受输入值
result = [str(0) for i in range(int(nums))]  # 初始化结果列表

for i in range(int(nums)):
    t_str = input()
    if ('P' in t_str) & ('T' in t_str):  # 排除包含多个 P、T的情况
        p_cnt, a_cnt, t_cnt = t_str.count('P'), t_str.count('A'), t_str.count('T')  # 统计PAT个数
        if (p_cnt == 1) & (t_cnt == 1) & (t_str.index('P') < t_str.index('T')):  # 只存在一个 P、T，并且P在T之后
            if (p_cnt * a_cnt * t_cnt != 0) & (p_cnt+a_cnt+t_cnt == len(t_str)):
                tmp = re.split('P|T', t_str)  # 切割字符串
                # 分支统计结果
                result[i] = 'YES' if tmp[0].count('A')*tmp[1].count('A') == tmp[2].count('A') else 'NO'
            else:
                result[i] = 'NO'
        else:
            result[i] = 'NO'
    else:
        result[i] = 'NO'

for val in result:
    print(val)