hdu 3336 Count the string(KMP+dp)

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

#define MAXN 200005//字符串长度
#define MOD 10007

char s[MAXN];
int dp[MAXN];

int _next[MAXN];

void GetNext(char t[]) //求next数组
{
int j,k,len;
j=0;//从0开始，首先求_next[1]
k=-1;//比较指针
_next[0]=-1;//初始值-1
len=strlen(t);
while(j<len)
{
if(k==-1||t[j]==t[k]) //指针到头了，或者相等
{
++j;
++k;
_next[j]=k;//此句可由优化替代
/*优化（求匹配位置时可用）
if(t[j]!=t[k])_next[j]=k;
else _next[j]=_next[k];
//*/
}
else k=_next[k];
}
}

int main()
{

int T;
int n;
int len;

int ans;

scanf("%d",&T);

while(T--)
{
scanf("%d",&n);
scanf("%s",s);

GetNext(s);
len=strlen(s);

ans=0;
dp[0]=0;

for(int i=1; i<=len; ++i) //i表示以i结尾的前缀字符串
{
dp[i]=dp[_next[i]]+1;//每个前缀增加个数1，dp[]为该前缀可以增加的个数
dp[i]%=MOD;
ans+=dp[i];
ans%=MOD;
}

printf("%d\n",ans);

}

return 0;
}