1040 有几个PAT Python实现

最新推荐文章于 2024-11-08 18:41:06 发布

原创最新推荐文章于 2024-11-08 18:41:06 发布 · 510 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#PAT练习 #Python语言 #程序设计

本文介绍了一种算法，用于计算给定字符串中特定子串‘PAT’出现的次数。通过遍历字符串找到所有可能组合的方法，并对结果进行优化处理以避免整数溢出。

1040 有几个PAT（25）（25 分）

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。

现给定字符串，问一共可以形成多少个PAT？

输入格式：

输入只有一行，包含一个字符串，长度不超过10^5^，只包含P、A、T三种字母。

输出格式：

在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大，只输出对1000000007取余数的结果。

输入样例：

APPAPT

输出样例：

问题分析：

在字符串中找字符A，并在该字符前面找字符P的数量，在该字符后面找字符T的数量，将两字符数量相乘得到该字符A可以构成的PAT数量，重复上述过程，最终结果对1000000007取余在输出。上述思路是博主从网上看到的，因为不知道哪篇博文是正真的原创，所以其代码就不贴出来了，大家可以自己搜索。

博主自己的思路是顺序寻找字符P，再从该位置顺序找字符A，再找字符T，是三层嵌套循环实现。无论哪种思路，都有相同测试超时。

程序（菜狗博主自己的，有超时）：

a = [i for i in input()]
x = 0
for i in range(len(a)):
    if a[i]=='P':
        for j in range(i+1,len(a)):
            if a[j]=='A':
                for k in range(j+1,len(a)):
                    if a[k]=='T':
                        x+=1
x = x%1000000007
print(x)