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;
}