PAT-1030-几个PAT-python(简洁)

• 题目描述

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

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

• 输入描述

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

• 输出描述

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

• 输入例子

APPAPT

• 输出例子

2

手动推算几个例子就会发现，这道题目有点类似于爬台阶。
可以通过每一个‘T’值把输入的字符串分割开来，上一个‘T’对应的所有组合，对下一个‘T’的计算至关重要。
具体思路，用一页ppt表示一下↓↓↓

s = input()
s = list(s) # 转成列表，后续需要为字符串里特殊位置赋整数型数值。
result = 0  # 存储结果
count_p = 0 # 记录当前位置之前，共有多少个P值
save_t = [] # 记录每一个T所能带来多少种可能的组合。
t_index = 0 # 记录'T'的下标
# 求得第一个T所对应的PAT个数之和。
for i in range(len(s)):
    if s[i] == 'P':
        count_p += 1
    if s[i] == 'A':
        s[i] = count_p
    if s[i] == 'T':
        for j in range(i):
            if s[j] != 'P' and s[j] != 'A' and s[j] != 'T':
                 result += s[j]
        save_t.append(result)
        t_index = i  # 保留上一个T值在字符串s中的下标，方便求取到第二个T之间，A值之和。
        break
# 接着求取剩余结果。
for p in range(i+1, len(s)):
    if s[p] == 'P':
        count_p += 1
    if s[p] == 'A':
        s[p] = count_p
    if s[p] == 'T':
        # 计算相邻的两个T之间所有A值之和。
        add_A = 0
        for q in range(t_index+1, p):
            if s[q] != 'P' and s[q] != 'A' and s[q] != 'T':
                add_A += s[q]
        Add = save_t[len(save_t)-1] + add_A
        result += Add 
        save_t.append(Add)
        t_index = p
                
print(result%1000000007)

踩坑点总结

• 一开始的做法是三个for循环暴力求解，10个样例都超时了，所以才找了规律。
• 注意题目输出要求，还有求个模。
• 注意定义的变量范围，过了范围，值就取不到了。
• 虽然是两个大for循环，实际上是一个。第二个是接着第一个for循环后面写的。所以不算复杂。