HDU - 3336 Count the string（KMP，前缀出现次数）

HDU - 3336 Count the string

#include<cstdio>
const int N = 200010;
char b[N];
int ne[N],m;
void get_next()
{
    for(int i=2,j=0;i<=m;i++)
	{
		while(j&&b[i]!=b[j+1]) j=ne[j];
		if(b[i]==b[j+1]) j++;
		ne[i]=j;
	}
}
int main()
{
	int T;scanf("%d",&T);
	while(T--)
	{
		scanf("%d%s",&m,b+1);
	    get_next();
	    int res=m;
	    for(int i=1;i<=m;i++)
	    	if(ne[i]>0)
	    		res++;
	    printf("%d\n",res%10007);
	}
	return 0;
}