hdoj3336（kmp算法next数组的应用）

题目链接：https://vjudge.net/problem/HDU-3336

题意：给定长为n(<=2e5)的字符串s，求s的每个前缀在s中出现的次数之和。

思路：

　　用dp[i]表示以s[i]为结尾的子串是s的某一种前缀的方案数，那么dp[i]=dp[nex[i]]+1，因为[nex[i]-(i-nex[i])+1 , nex[i]] 和 [nex[i]+1 , i] 是一样的，所以计算后者时只用把前者+1即可（+1是因为以[1,i]为前缀的个数加一）。

AC代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
 
const int maxn=2e5+;
const int MOD=1e4+;
int T,n,ans,nex[maxn],dp[maxn];
char s[maxn];
 
void get_next(){
    int j;
    j=nex[]=-;
    for(int i=;i<n;++i){
        while(j>-&&s[i]!=s[j+]) j=nex[j];
        if(s[i]==s[j+]) ++j;
        nex[i]=j;
    }
}
 
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%d%s",&n,s);
        get_next();
        ans=;
        for(int i=;i<=n;++i){
            dp[i]=(dp[nex[i-]+]+)%MOD;
            ans=(ans+dp[i])%MOD;
        }
        printf("%d\n",ans);
    }
    return ;
}