[51nod 1304]字符串的相似度

题目描述

我们定义2个字符串的相似度等于两个串的相同前缀的长度。例如 “abc” 同 “abd” 的相似度为2，”aaa” 同 “aaab” 的相似度为3。

给出一个字符串S，计算S同他所有后缀的相似度之和。例如：S = “ababaa”，所有后缀为：

ababaa 6

babaa 0

abaa 3

baa 0

aa 1

a 1

S同所有后缀的相似度的和 = 6 + 0 + 3 + 0 + 1 + 1 = 11

扩展KMP

做一遍扩展KMP然后把next都加起来。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=1000000+10;
int next[maxn];
char s[maxn];
int i,j,k,l,t,n,m;
ll ans;
int main(){
scanf("%s",s+1);
n=strlen(s+1);
j=2;
while (s[1+next[2]]==s[2+next[2]]) next[2]++;
fo(i,3,n){
if (i>j+next[j]-1){
while (s[1+next[i]]==s[i+next[i]]) next[i]++;
}
else{
next[i]=min(next[i-j+1],j+next[j]-i);
while (s[1+next[i]]==s[i+next[i]]) next[i]++;
}
if (i+next[i]>j+next[j]) j=i;
}
fo(i,2,n) ans+=(ll)next[i];
ans+=(ll)n;
printf("%lld\n",ans);
}