bzoj3238 [Ahoi2013]差异 后缀自动机

题意：给出一个串，求其中任意两个字串的lcp的总和。

我们可以对于这个串建一颗后缀自动机，实际上，他的parent边树就是一颗后缀树，我们在后缀树上统计答案，设f表示right集合的大小，可以理解为后缀树的siz，然后w是f的前缀和。mx按照基数排序，然后扫一遍就可以统计答案了。

 #include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#include<map>
using namespace std;
int n,m;
typedef long long ll;
const int N=1e6+5;
map<int,int>ch[N];
int mx[N],fa[N],f[N],g[N],b[N],c[N];
int last,cnt;
char s[N];
ll ans;
inline void ins(int x)
{
    int p,q,np,nq;//p:status  0;
    p=last;       //np:newnode
    last=np=++cnt;
    mx[np]=mx[p]+1;
    g[np]=f[np]=1;
    for(;!ch[p][x]&&p;p=fa[p])ch[p][x]=np;
    if (!p)fa[np]=1;
    else
    {              //nq:son[q]
        q=ch[p][x];//q:laststatus
        if (mx[q]==mx[p]+1)fa[np]=q;
        else
        {
            nq=++cnt;
            mx[nq]=mx[p]+1;
            ch[nq]=ch[q];
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            for(;ch[p][x]==q;p=fa[p])ch[p][x]=nq;
        }
    }
}
inline ll solve()
{
    fo(i,1,cnt)b[mx[i]]++;
    fo(i,1,n)b[i]+=b[i-1];
    fo(i,1,cnt)c[b[mx[i]]--]=i;
    fd(i,cnt,1)f[fa[c[i]]]+=f[c[i]];
    ll ans=0;
    fo(i,1,cnt)
    {
        ans+=1ll*g[fa[i]]*f[i]*mx[fa[i]];
        g[fa[i]]+=f[i];
    }
    return ans;
}
int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    last=cnt=1;
    fo(i,1,n)ins(s[i]-'a');
    ans=0;
    fo(i,1,n)ans+=1ll*i*(n-1);
    printf("%lld\n",ans-solve()*2);
    return 0;
}