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
输出样例:
2
问题分析:
在字符串中找字符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)