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;
}