【KMP】 HDOJ 3336 Count the string

简单的KMP题目，先求出next数组，累加每个数通过next数组到0的次数，最后加上字符串的长度就是答案喽。。

#include <iostream>  
#include <sstream>  
#include <algorithm>  
#include <vector>  
#include <queue>  
#include <stack>  
#include <map>  
#include <set>  
#include <bitset>  
#include <cstdio>  
#include <cstring>  
#include <cstdlib>  
#include <cmath>  
#include <climits>  
#define maxn 200005
#define eps 1e-6 
#define mod 10007 
#define INF 99999999  
#define lowbit(x) (x&(-x))  
#define lson o<<1, L, mid  
#define rson o<<1 | 1, mid+1, R  
typedef long long LL;  
using namespace std;  

int len;
char p[maxn];
int next[maxn];
int sum[maxn], vis[maxn];

void get(void)
{
	int i, j;
	next[0]=next[1]=0;
	for(i=1;i<len;i++){
		j=next[i];
		while(j && p[i]!=p[j]) j=next[j];
		next[i+1]= p[i]==p[j] ? j+1 : 0;
	}
}
void solve(void)
{
	memset(vis, 0, sizeof vis);
	memset(sum, 0, sizeof sum);
	for(int i=1;i<=len;i++){
		if(next[i]) sum[i]=sum[next[i]]+1;
	}
}
int main(void)
{
	int _, ans, i;
	while(scanf("%d",&_)!=EOF){
		while(_--){
			scanf("%d",&len);
			scanf("%s", p);
			get();
			solve();
			ans=len%mod;
			for(i=1;i<=len;i++)
				ans+=sum[i], ans%=mod;
			printf("%d\n", ans);
		}
	}
	return 0;
}