PAT甲级1093 Count PAT

pnum【i】记录对于下标为 i 的位置，其左边的 ’ P ’ 个数；
tnum【i】记录对于下标为 i的位置，其右边’ T '的个数；
然后对于每个’A’，以当前A为中心的PAT组合个数就是pnum【i】*tnum【i】，对每个A计算该值，然后累加，输出最后结果即可

#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
int main()
{
    string s;
    cin>>s;
    vector<int> pnum,tnum;
    vector<int> calculate;//记录需要处理的A下标
    pnum.resize(s.size());
    tnum.resize(s.size());
    int p=0,t=0;
    for(int i=0;i<s.size();i++)
    {
        if(s[i]=='P')p++;
        else if(s[i]=='A')calculate.push_back(i);
        pnum[i]=p;
    }
    for(int i=s.size()-1;i>=0;i--)
    {
        if(s[i]=='T')t++;
        tnum[i]=t;
    }
    long long result=0,temp;
    for(int i=0;i<calculate.size();i++)
    {
        temp=pnum[calculate[i]]*tnum[calculate[i]];
        result+=temp;
        result%=1000000007;
    }
    cout<<result;
    return 0;
}