C++/Python PAT乙级 1040 有几个PAT（25）

题目描述

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位§,第4位(A),第6位(T)；第二个PAT是第3位§,第4位(A),第6位(T)。
现给定字符串，问一共可以形成多少个PAT？

输入描述:

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

输出描述:

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

输入例子:

APPAPT

输出例子:

2

C++

#include <iostream>
#include <cmath>
#include <cstring>
using namespace std;

int main()
 {
    string str;
    int n=1000000007;
    int p=0,pa=0,pat=0;
    cin >> str;
    int num=str.length();
    for (int i=0;i<num;i++)
    {
        if(str[i]=='P') p++;
        else if(str[i]=='A') {pa+=p;pa%=n;}//要加上前面的‘p’的个数
        else if(str[i]=='T') {pat+=pa;pat%=n;}//加上前面‘pa’的个数
    }
    cout << pat;
    return 0;
 }

Python

s=list(input())
p,pa,pat=0,0,0
for i in range(len(s)):
    if s[i]=='P':
        p+=1
    elif s[i]=='A':
        pa+=p
    elif s[i]=='T':
        pat+=pa
print(pat%1000000007 )