PAT (Advanced Level) 1093. Count PAT's (25)

预处理每个位置之前有多少个P，每个位置之后有多少个T。

对于每个A，贡献的答案是这个A之前的P个数*这个A之后T个数。

#include<cstdio>
#include<cstring>

long long MOD=1e9+;
const int maxn=1e5+;

long long dp1[maxn],dp2[maxn];
char s[maxn];

int main()
{
    scanf("%s",s);
    memset(dp1,,sizeof dp1);
    if(s[]=='P') dp1[]=;
    for(int i=;s[i];i++)
    {
        dp1[i]=dp1[i-];
        if(s[i]=='P') dp1[i]++;
    }
    memset(dp2,,sizeof dp2);
    int len=strlen(s);
    if(s[len-]=='T') dp2[len-]=;
    for(int i=len-;i>=;i--)
    {
        dp2[i]=dp2[i+];
        if(s[i]=='T') dp2[i]++;
    }
    long long ans=;
    for(int i=;s[i];i++)
    {
        if(s[i]=='A')
        ans=(ans+dp1[i-]*dp2[i+])%MOD;
    }
    printf("%lld\n",ans);
    return ;
}