有几个PAT

描述

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

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

 

输入

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

 

输出

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

 

样例输入

APPAPT

样例输出

2

#include <bits/stdc++.h>  //动态规划判断A之前有多少个P 和A之后有多少个T
                          //求后面这个反过来求T之前有多少个A
using namespace std;
const int N=1e5+5;
typedef long long ll;
int main()
{
    ios::sync_with_stdio(false);
    string ss;
    cin>>ss;
    ll dp1[N],dp2[N];
    memset(dp1,0,sizeof(dp1));
    memset(dp2,0,sizeof(dp2));
    if(ss[0]=='P')
        dp1[0]=1;
    else dp1[0]=0;
    ll num=0;ll u=0;
    for(int i=1;i<ss.size();i++){   //这个位置的A前面又多少个P
        if(ss[i]=='A'){
            dp1[i]=dp1[u]+num;
            num=0;
            u=i;   //u  标记前面的A
//            printf("%d %d
",i,dp1[i]);
        }
        if(ss[i]=='P') num++;
    }
    ll len=ss.size()-1;
    if(ss[len]=='T')
        dp2[len]=1;
    else dp2[len]=0;
    num=0;u=len;
    for(int i=ss.size()-2;i>=0;i--){  //这个位置的A后面有多少个T
        if(ss[i]=='A'){
            dp2[i]=dp2[u]+num;
            num=0;
            u=i;
        }
        if(ss[i]=='T'){
            num++;
        }
    }
//    printf("%d   
",dp1[3]);
//    for(int i=0;i<=len;i++){
//        printf("dp1[i]=%d  dp2[i]=%d
",dp1[i],dp2[i]);
//    }
    ll sum=0;
    for(int i=0;i<=len;i++){
        if(dp1[i]!=0&&dp2[i]!=0){
            sum=(sum+dp1[i]*dp2[i])%1000000007;
        }
    }
    printf("%lld
",sum);
    return 0;
}