HDU 3336 Count the string（KMP）

题目链接：HDU 3336 Count the string（KMP）

题目大意：

求给定字符串前缀重复次数的和。

每个前缀本身重复一次，

再加上所有后缀与前缀匹配的次数就是答案了。

dp的方法没有看懂，这里学习了KIDxの博客 的文章。

把每个递增的next的最后一个值加上，就是当前所有匹配后缀的数目。

盗图。

例如 next [ 5 ] = 2，next [ 6 ] = 1,

5前面的串 abcab ， 最长的匹配后缀的长度为 2 ， 也就是说 ab 和 ab ，既然 ab 和 ab 匹配， 那么 a 和 a 自然也匹配，因此 这个最长匹配后缀就囊括了该位置之前匹配后缀的数目。

所以对于这个样例 。

我们ans先加上字符串的长度， 也就是所有前缀的数目。

然后加上所有递增的next的最后一个位置的值。 next [ 5 ] , next[ 6 ] , next [ 12 ](最后一个位置), 就是所求的结果了。

源代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 200000 + 10;
char s[maxn];
int nextv[maxn];
int len;
void getnext(){
	int i=0,j=-1;
	nextv[0] = -1;
	while(i<len){
		if(j == -1 || s[i] == s[j])
			nextv[++i] = ++j;
		else
			j = nextv[j];
	}
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		int n;
		scanf("%d%s",&n,s);
		len = strlen(s);
		getnext();
		long long ans = nextv[len] + len; //先加上最后的 nextv ， 以及字符串的长度（前缀数目）
		// for(int i=0;i<=len;i++)
			// printf("%d ",nextv[i]);
		// cout<<endl;
		for(int i=0;i<len;i++){
			if(nextv[i] > 0 && nextv[i]+1 != nextv[i+1]){ //找到所以最大后缀匹配的位置
				ans += nextv[i];
				ans %= 10007;
			}
		}
		
		printf("%lld\n",ans%10007);
	}
	return 0;
}