hdu3336 Count the string(kmp+dp)

题意：给出一个字符串，求所有前缀串出现的总次数。

abab 的前缀串有：a  ab  aba  abab

其中a出现2次

总共出现6次。

思路：先是用KMP求出第i个字符重复的最长前缀的结束位置

           dp[i]代表长为i的前缀串（数量1）与其有重复前缀数量。

#include<iostream>#include<queue>using namespace std;const int N = 222222,M = 10007;int dp[N],next[N];char s[N];void get_next(int n){for (int i=2,j=0;i<=n;i++){//cout<<s[j+1]<<s[i]<<endl;while (j && s[j+1]!=s[i]) j=next[j];if (s[j+1] == s[i]) j++;next[i]=j;}}int main(){freopen("in","r",stdin);int t,n;cin>>t;while (t--){scanf("%d",&n);scanf("%s",s+1);s[0]='#';get_next(n);for (int i=1;i<=n;i++) dp[i]=1;int ans=0;//for (int i=1;i<=n;i++) cout<<next[i]<<' ';cout<<endl;for (int  i=1;i<=n;i++){dp[i]=dp[next[i]]+1;ans+=dp[i];ans%=M;}printf("%d\n",ans);}}