hdu 3336 Count the string

题目：

链接：http://acm.hdu.edu.cn/showproblem.php?pid=3336

算法：

字符串KMP。好的，先来复习一下在KMP中next数组是怎么求的？next[i]表示在i位置前面的串的前缀和后缀相等的长度。例如：abab:-1 0 0 1;其中b位置处next为1，表示串aba中的前缀a和后缀a相等，长度为1。

首先，我们规定第一个字符的模式值为-1，若S[j] == S[k],那么就有next[j+1] = next[k+1]；若不相等，即前后缀匹配失败，k = next[k]。

思路：

利用next数组求出每个prefix的匹配串个数。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int t,n;
char s[200020],s1[200020];
int next[200020];

void get_nextval()
{
int i,j;
i=-1;
j=0;
next[0]=-1;
while (j<n)
{
if(i==-1 || s[i]==s[j])
{
j++; i++;
next[j]=i;
}
else
i=next[i];
}
}

int main()
{
//freopen("input.txt","r",stdin);
int sum,j;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
getchar();
gets(s);
get_nextval();
sum = n;    //一定存在的串个数
for(int i=1; i<=n; i++)
{
for(j=next[i]; j; sum++)
j = next[j];
sum %= 10007;
}
printf("%d\n",sum);
}
return 0;
}