51nod 1304 字符串的相似度 拓展kmp

题意

我们定义2个字符串的相似度等于两个串的相同前缀的长度。例如 “abc” 同 “abd” 的相似度为2，”aaa” 同 “aaab” 的相似度为3。
给出一个字符串S，计算S同他所有后缀的相似度之和。
1 <= L <= 1000000

分析

看完题后：哈哈哈这不是sa模板题吗。花十分钟敲完后发现居然T了。。。
后来听说是exkmp的模板题，就又跑去学了一发exkmp。
直接把next数组全部加起来就好了。
一开始打exkmp的时候把pos的初值设为了1，然后就变成了大暴力。。。实际上赋为2就可以。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;typedef long long LL;const int N=1000005;int n,next[N];char s[N];void get_next(){    int mx=0,pos=0;    next[1]=n;    for (int i=2;i<=n;i++)    {        if (mx>=i) next[i]=min(mx-i+1,next[i-pos+1]);        while (i+next[i]<=n&&s[i+next[i]]==s[next[i]+1]) next[i]++;        if (i+next[i]-1>mx) mx=i+next[i]-1,pos=i;    }}int main(){    scanf("%s",s+1);    n=strlen(s+1);    get_next();    LL ans=0;    for (int i=1;i<=n;i++) ans+=next[i];    printf("%lld",ans);    return 0;}