hdu 3336 count the string（KMP+dp）

题目大意：

   给你一个字符串，问这个字符串的所有前缀在这个字符串中出现的次数。

解题思路：

  dp【i】表示以i-1结尾的前缀的个数。

  dp[j]=dp[f[j]]+1;f[]为kmp中的失配函数。

  如果不能理解手动模拟下。

我的代码：

#include <cstdio>
#include <cstring>
using namespace std;
void getFail (char *P,int *f){ //构建失配函数 f为失配函数 P为模板串
  int m=strlen(P);
  f[0]=f[1]=0;
  for (int i=1;i<m;i++){
    int j=f[i];
    while (j&&P[i]!=P[j]) j=f[j];
    f[i+1]=P[i]==P[j]?j+1:0;
  }
}
int ans;
char T[200005];
int f[200005];
int dp[200005];
int main (){
  int t;scanf("%d",&t);
  while (t--){
    int n;
    scanf("%d",&n);
    scanf("%s",T);
    getFail(T,f);
    int sum=0;dp[0]=0;
    for (int i=1;i<=n;i++){
      dp[i]=dp[f[i]]+1;
      sum=(sum+dp[i])%10007;
    }
    printf("%d\n",sum);
  }
  return 0;
}