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=;
 const int MOD=;
 int dp[MAXN];
 char str[MAXN];
 int next[MAXN];

 void getNext(char *p)
 {
     int j,k;
     j=;
     k=-;
     next[]=-;
     int len=strlen(p);
     while(j<len)
     {
         if(k==-||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[]=;
         int ans=;
         for(int i=;i<=n;i++)
         {
             dp[i]=dp[next[i]]+;
             dp[i]%=MOD;
             ans+=dp[i];
             ans%=MOD;
         }
         printf("%d\n",ans);
     }
     return ;
 }