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