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

转载于 2016-07-04 20:40:00 发布 · 80 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zufezzt/p/5641654.html

本文介绍了一种用于字符串处理的算法，该算法可以计算特定字符组合在字符串中的出现次数。通过对字符串进行预处理，记录每个位置前后的特定字符数量，进而高效地计算目标组合的总出现次数。

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

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

#include<cstdio>
#include<cstring>

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

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

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