HDU 3336 Count the string

KMP中next数组的妙用+DP

又是一道问前缀的题，显然是要用KMP，但是如果枚举每一个前缀，再到主串里跑KMP，会TLE，那么考虑DP，想到next数组的定义，并不用题目中的方法对于每一个前缀求在主串中出现了多少会，而是对于每一个前缀，求出以遍历过得前缀在该前缀中出现了多少次，所以dp[i]=dp[next[i]]+1。对于每一个前缀来说，在其中新出现的的前缀一定在，next中。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cmath>
using namespace std;
const int MOD=10007,MAXN=200007;
int init(){
	int rv=0,fh=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-') fh=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		rv=(rv<<1)+(rv<<3)+c-'0';
		c=getchar();
	}
	return fh*rv;
}
int T,n,nxt[MAXN],dp[MAXN],ans;
char s[MAXN];
void getnxt(){
	nxt[0]=-1;
	int k=-1,j=0;
	while(j<n){
		if(k==-1||s[k]==s[j]) nxt[++j]=++k;
		else k=nxt[k];
	}
}
int main(){
	freopen("in.txt","r",stdin);
	T=init();
	while(T--){
		n=init();
		scanf("%s",s);
		getnxt();
		for(int i=1;i<=n;i++){
			dp[i]=dp[nxt[i]]+1%MOD;
			(ans+=dp[i])%=MOD; 
		}
		cout<<ans<<endl;
		ans=0;
	}
	
	fclose(stdin);
	return 0;
}