HDU 3336 Count the string KMP+DP

传送门http://acm.hdu.edu.cn/showproblem.php?pid=3336

大意：数前缀匹配了多少次如：s: "abab"
前缀为: "a", "ab", "aba", "abab"      a匹配了2次，ab两次,aba一次，abab一次，故答案为6

KMP的应用。

Orz发明这个算法的人。

 

 

状态转移函数 F[i]保证了s[1 .. i]中 s[i - F[i] + 1 .. i]与 s[1 .. F[i]]是相等的。

 

设数组cnt：以 i结尾的串中所有前缀的计数和

则状态转移方程：  cnt[i]= cnt[ F[i] ] + 1;

 

#include<cstdio>
#include<cstring>
const int mod=10007;
const int MAXN=200000+10;
char p[MAXN];
int f[MAXN],n,cnt[MAXN];
void getfail()
{
    f[0]=f[1]=0;
    int j;
    for(int i=1;i<n;i++)
    {
        j=f[i];
        while(j && p[i]!=p[j])
            j=f[j];

        if(p[i]==p[j])
            j++;

        f[i+1]=j;
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int ans=0;
        scanf("%d",&n);
        scanf("%s",p);
        getfail();
        memset(cnt,0,sizeof(cnt));
        for(int i=1;i<=n;i++)
        {
            cnt[i]=cnt[ f[i] ] +1;
            ans=(ans+cnt[i])%mod;
        }
        printf("%d\n",ans);
    }
}