1040 有几个PAT (25分)

 

1040 有几个PAT (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

 

时间限制: 150 ms

内存限制: 64 MB

代码长度限制: 16 KB

 

对每一个A统计左边P的个数乘以右边T的个数（事先用数组保存好每一位P的个数和T的个数会比较省时）

#include<iostream>
#include<vector>
using namespace std;

#define MaxNum 1000000007

int main(){
    string str;
    cin>>str;
    int n=str.length();
    if(n==0){cout<<"0";return 0;}
    vector<int> p(n,0);
    vector<int> t(n,0);
    if(str[0]=='P')p[0]++;
    for(int i=1;i<n;i++){
        if(str[i]=='P')p[i]=p[i-1]+1;
        else p[i]=p[i-1];
    }
    if(str[n-1]=='T')t[n-1]++;
    for(int i=n-2;i>=0;i--){
        if(str[i]=='T')t[i]=t[i+1]+1;
        else t[i]=t[i+1];
    }
    int res=0;
    for(int i=0;i<n;i++)
    {
        if(str[i]=='A')
        {
            res=(res+p[i]*t[i])%MaxNum;
        }
    }
    cout<<res;
    return 0;
}