HDU 3336 Count the string 【KMP】【dp】

题目：点击打开链接

题意：给定一个字符串，求出字符串中对应各个前缀，在字符串中出现的次数和。

分析：显然用kmp，得出next数组后观察容易得到，每一个字符的next为要跳转的位置，则当每次跳转后都要给先前会跳转的字符次数加一，故可以用dp的思想，根据当前的次数和，下一个（即序号前一个）的次数等于当前位置的次数加一，将所有的次数相加就可以得到答案了。

代码：

#include <stdio.h>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
#include <set>
#include <vector>
#include <map>
#define PR pair<int,int>
#define MP make_pair
#define fi first
#define se second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define sqr(x) ((x)*(x))
#define ll long long
#define ull unsigned __int64
const ll INF = 1e18;
const int inf=0x3f3f3f3f;
const int M=200010;
const int N=200100;
const int MOD=10007;
const double eps=1e-10;
const double pi=acos(-1.0);
using namespace std;

int nxt[N],n;
ll sum[N];
char a[N];
void cal_nxt()
{
    nxt[0]=-1;
    for(int i=1;i<n;i++)
    {
        int j=nxt[i-1];
        while(j>=0&&a[i]!=a[j+1]) j=nxt[j];
        nxt[i]=a[i]==a[j+1]?j+1:-1;
    }
}

int main()
{
    int casT,i;
    scanf("%d",&casT);
    while(casT--)
    {
        memset(sum,0,sizeof(sum));
        int ans=0;
        scanf("%d%s",&n,a);
        cal_nxt();
        for(i=n-1;i>=0;i--)
        {
            if(nxt[i]!=-1) sum[nxt[i]]+=sum[i]+1;
            ans=(ans+sum[i]+1)%MOD;
        }
        printf("%d\n",ans);
    }
}