bzoj 3238: [Ahoi2013]差异 后缀自动机

题意

n<=500000

分析

题目其实就是要求sigma(1<=i,j<=n) lcp(Ti,Tj)
我们将字符串建立一棵后缀自动机，那么两个前缀的lcp就是其在parents树上的lca的最长子串。
树形dp一下即可。
这是本蒟蒻第一道SAM的题目，在此发表一下对SAM的理解。
代码一点都不复杂，只是要看懂确实要花蛮大的功夫，从cls60+页的课件就可以看出来。
首先SAM上每个节点都包含了一个right集，其right集就等于其所有子节点的right集的并集。那么一个right集加一个长度唯一表示原串的一个子串。对于每一个right集都有一个长度区间[min,max]，所有min<=x<=max的x对于该right集都是一个符合条件的长度。
差不多就是这样了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define N 500005
#define LL long long
using namespace std;

int n,l[N*2],ch[N*2][30],fa[N*2],f[N*2],w[N*2],b[N*2],c[N*2],last,root,cnt;
char s[N];

void ins(int x)
{
    int p,q,np,nq;
    p=last;last=np=++cnt;l[np]=l[p]+1;w[np]=f[np]=1;
    for (;!ch[p][x]&&p;p=fa[p]) ch[p][x]=np;
    if (!p) fa[np]=root;
    else
    {
        q=ch[p][x];
        if (l[q]==l[p]+1) fa[np]=q;
        else
        {
            nq=++cnt;l[nq]=l[p]+1;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            for (;ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
        }
    }
}

LL solve()
{
    for (int i=1;i<=cnt;i++) b[l[i]]++;
    for (int i=1;i<=n;i++) b[i]+=b[i-1];
    for (int i=1;i<=cnt;i++) c[b[l[i]]--]=i;
    for (int i=cnt;i>=1;i--) f[fa[c[i]]]+=f[c[i]];
    LL ans=0;
    for (int i=1;i<=cnt;i++)
    {
        ans+=(LL)w[fa[i]]*f[i]*l[fa[i]];
        w[fa[i]]+=f[i];
    }
    return ans;
}

int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    last=cnt=root=1;
    for (int i=1;i<=n;i++) ins(s[i]-'a');
    LL ans=0;
    for (int i=1;i<=n;i++)
        ans+=(LL)i*(n-1);
    printf("%lld",ans-solve()*2);
    return 0;
}