HDU 3336(Count the string)DP+KMP

题目要求字符串所有前缀在字符串出现的次数之和，次数包括重叠部分（例如ababa,前缀aba出现次数为2次）

朴素思想，枚举前缀，各自KMP算个数，题目数据量较大，不允许这样。

前后缀问题一般会想到利用NEXT数组的性质。

例如abababa，NEXT:0012345，设dp[i]为第i字符为止，前缀(1~i)出现的次数，

dp[i]=dp[nextp[i]]+1

代码

#include<iostream>#include<cstring>using namespace std;#define MOD 10007int nextp[200005],dp[200005];char s[200005];void Getnext(int len){    int i,j;    i=0,j=-1;    nextp[0]=-1;    while(i<len)    {        if(j==-1 || s[i]==s[j])        {            i++;            j++;            nextp[i]=j;        }        else j=nextp[j];    }}int main(){    int T;    int i,j,k,ans,n;    cin>>T;    while(T--)    {        cin>>n;        cin>>s;        Getnext(n);        ans=0;        memset(dp,0,sizeof(dp));        for(i=1;i<=n;i++)        {            dp[i]=(dp[nextp[i]]+1)%MOD;            ans=(ans+dp[i])%MOD;        }        cout<<ans<<endl;    }    return 0;}