hdu3336 Count the string kmp+dp

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3336

很容易想到用kmp

这里是next数组的应用

定义dp[i]表示以s[i]结尾的前缀的总数

那么dp[i]=dp[next[i]]+1;

代码：

#include<stdio.h>
#include<string.h>
const int MAXN=200020;
const int MOD=10007;
int dp[MAXN];
char str[MAXN];
int next[MAXN];

void getNext(char *p)
{
    int j,k;
    j=0;
    k=-1;
    next[0]=-1;
    int len=strlen(p);
    while(j<len)
    {
        if(k==-1||p[j]==p[k])
        {
            j++;
            k++;
            next[j]=k;
        }
        else k=next[k];
    }
}
int main()
{
    int T;
    int n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        scanf("%s",&str);
        getNext(str);
        dp[0]=0;
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            dp[i]=dp[next[i]]+1;
            dp[i]%=MOD;
            ans+=dp[i];
            ans%=MOD;
        }
        printf("%d
",ans);
    }
    return 0;
}