HDU 3336 KMP 求所有前缀在母串中出现的次数

题意：

T个测试数据

n表示母串长度

下面给定母串

问母串的所有前缀，在母串中出现的次数和

而所有前缀和 的状态转移方程很显然是 dp[i] = dp[ f[i] ] + 1;

则 ans = 求和 dp

#include <stdio.h>#include <string.h>char P[200100];//从0开始存int f[200100];//记录P的自我匹配int len;void getFail(){f[0]=f[1]=0;for(int i=1;i<len;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 dp[200100], ans;int main(){int T;scanf("%d",&T);while(T--){scanf("%d",&len);scanf("%s",P);int ans = 0;getFail();memset(dp, 0, sizeof(dp));dp[0] = 0;ans = 0;for(int i = 1;i <= len;i++){dp[i] = (dp[f[i]] + 1)%10007;ans += dp[i];if(ans>10007)ans %= 10007;}printf("%d\n",ans);}return 0;}/*994abab6ababab*/