题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3336
解题思路:
先把原数组构造失配数组
fail[i] 表示0~i-1最长相同前后缀
首先我们先令ans = len,也就是先把每个前缀本身加上去
我们可以保证在fail数组所得到的后缀不会与这些重合,不然那叫前缀。
接下去遍历fail[2]~fail[len]
对于每个fail我们求出所有以i-1为后缀结尾的后缀
首先当前fail[i]不为0的话,一定存在0 ~ x == i-1-x ~ i-1
并且我们保证这个后缀之前没有被计数,因为之前没有得到过i-1为结尾的后缀
我们要得到所有以i-1为后缀的要怎么做,就是沿着失配边回去,看fail[fail[i]]是否大于0,大于0证明还可以从这个i-1-x~i-1再抠出一个以i-1结尾的更短的后缀,再看fail[fail[fail[i]]]直到这个值不再大于0.
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mod 10007
using namespace std;
const int N = 2e5+5;
char s[N];
int fail[N];
void kmp(int len)
{
fail[0] = -1;
for (int i=0,j=-1;i<len;){
if (j==-1||s[i]==s[j]){
i++;j++;
fail[i] = j;
}
else j = fail[j];
}
int ans = len%mod;
for (int i=2;i<=len;i++){
if (fail[i]>0) ans++;
int now = fail[i];
while (fail[now]>0){
ans++;
now = fail[now];
}
}
printf("%d\n",ans%mod);
}
int main()
{
int T;
scanf("%d",&T);
while (T--){
int n;
scanf("%d",&n);
scanf("%s",s);
kmp(n);
}
return 0;
}
也不知道我这么想是不是复杂了,以后如果从其他题感受到了更简单的方法就回来改一下