hdu 3336 count the string（KMP+dp）

题意：

求给定字符串，包含的其前缀的数量。

分析：

就是求所有前缀在字符串出现的次数的和，可以用KMP的性质，以j结尾的串包含的串的数量，就是next[j]结尾串包含前缀的数量再加上自身是前缀，dp[i]表示以i为结尾包含前缀的数量，则dp[i]=dp[next[i]]+1，最后求和即可。

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<11
#define All 1,N,1
#define read freopen("in.txt", "r", stdin)
#define N 200010
const ll  INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
const int mod =  10007;
int dp[N],f[N],n;
char str[N];
void getnext(){
    int i=0,j=-1;
    f[0]=-1;
    while(i<n){
        if(j==-1||str[i]==str[j])
        {
            i++;j++;
            f[i]=j;
        }
        else j=f[j];
    }
}
void solve(){
    getnext();
        //cout<<1<<endl;
    memset(dp,0,sizeof(dp));
    int total=0;
    for(int i=1;i<=n;++i){
        dp[i]=(dp[f[i]]+1)%mod;
        total=(total+dp[i])%mod;
    }
    printf("%d\n",total);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        scanf("%s",str);
        solve();
    }
return 0;
}