p数组统计当前位置之前有多少个P
t数组统计当前位置后面有多少个T
当遇到A时就计算两者的乘积再加上之前的结果再取模
存在大量重复计算,所以可以视为简单动态规划题
#include<bits/stdc++.h>
char str[100005];
int p[100005]={0};
int t[100005]={0};
int main(){
scanf("%s",str);
int len=strlen(str);
for(int i=0;i<len;i++){
if(i==0){
if(str[i]=='P') p[i]=1;
if(str[len-1-i]=='T') t[len-1-i]=1; //赋值符号写成了等号 调试半天
continue;
}
if(str[i]=='P')
p[i]=p[i-1]+1;
else
p[i]=p[i-1];
if(str[len-i-1]=='T')
t[len-i-1]=t[len-i]+1;
else
t[len-i-1]=t[len-i];
}
int ans = 0;
for(int i=0;i<len;i++){
if(str[i]=='A') {
ans=(ans+p[i]*t[i])%1000000007;
}
}
printf("%d\n",ans);
return 0;
}