【PAT甲级】1093 Count PAT's (25 分)

题意：

输入一行由大写字母'P','A','T'，组成的字符串，输出一共有多少个三元组"PAT"（相对顺序为PAT即可），答案对1e9+7取模。

代码：

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
const long long mod = 1e9+7;
string s;
long long nump[100007],numpa[100007];
long long sum[100007];
int main(){
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin>>s;
int n=s.size();
for(int i=0;i<n;++i){
nump[i]=nump[i-1];
if(s[i]=='P'){
++nump[i];
nump[i]%=mod;
}
numpa[i]=numpa[i-1];
if(s[i]=='A'){
numpa[i]+=nump[i];
numpa[i]%=mod;
}
sum[i]=sum[i-1];
if(s[i]=='T'){
sum[i]+=numpa[i];
sum[i]%=mod;
}
}
cout<<sum[n-1];
return 0;
}