HDU3336——KMP算法

题意是问所有前缀出现的次数和，mod10007；

想一想next数组代表什么意思，是从当前失配位置走到上一个匹配位置的后面，next[i]的值说明以当前位置为结尾，长度为next[i]的后缀，与以开头元素为起始，长度为next【i】

的前缀是相同的，那么方法就很容易了，对于每个j = i,沿着next【j】往前走，到0为止，记录走过的次数，就是前缀的出现次数。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<map>
#include<vector>
#include<cmath>
#include<set>
using namespace std;
const int maxn = 1e6+5;
const int mod = 10007;
char s[maxn];

vector<int> find_sub(char *pattern)//预处理next
{
int n = strlen(pattern);
vector<int> next(n+1,0);
for(int i = 1; i < n; ++i)
{
int j = i;
while(j > 0){
j = next[j];
if(pattern[j] == pattern[i]){

next[i+1] = j+1;
break;
}
}
}
return next;
}

int main()
{
// freopen("in","r",stdin);
int T;
scanf("%d",&T);
while(T--)
{
int n;
scanf("%d",&n);
scanf("%s",s);
vector<int> v = find_sub(s);
int ans = 0;

for(int i = 0; i < v.size(); ++i)
{
int j = i;
while(j){
ans = (ans+1)%mod;
j = v[j];

}
}
printf("%d\n",ans);
}

}