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