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