hdu3336(next数组的应用）

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=200002;
int Next[200002];
char t[200002];
int tlen;
int num[200002];
void MakeNext()
{
int j,k;
j=0,k=-1;
Next[0]=-1;
while(j<tlen)
{
if(k==-1||t[j]==t[k])
Next[++j]=++k;
else k=Next[k];
}
}
//next数组返回的是如果模式串位置k处字符与主串字符不匹配
//则主串该处的字符应该与模式串以next[k]为下标的字符比较
//next[k]其实也是k处前与主串匹配的字符串的长度
int main()
{

int n;
scanf("%d",&n);
while(n--)
{
scanf("%d",&tlen);
scanf("%s",t);
MakeNext();
long long sum=0;
memset(num,0,sizeof(num));
for(int i=1;i<=tlen;i++)
{
int tmp=Next[i];
while(tmp!=0)
{
num[tmp]++;//字符串长度为tmp的串匹配次数++
tmp=Next[tmp];
}
}
for(int i=0;i<tlen;i++)
{
sum=(sum+num[i]+1)%10007;//加上自己与自己匹配的1次
}
cout<<sum<<endl;
}
return 0;
}