【SAM】[AHOI2013]差异

题目

https://www.luogu.com.cn/problem/P4248

思路

考虑建SAM
根据定义，两个点在SAM上的parent树的LCA即为其LCS的距离，由于人以两个子串的LCS=任意两个字串的LCP，所以不用反过来插入(当然反过来也行）
所以我们在SAM上做DP，维护一下子树的size，计算这条边经过多少次

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e6+77;
char st[N];
int n;
ll ans;
struct SAM
{
    int tot,las,ch[N][27],fail[N],ml[N],siz[N],c[N],a[N];
    void ins(int c)
    {
        int p=las,np=las=++tot;
        ml[np]=ml[p]+1; siz[np]=1;
        while(p&&!ch[p][c]) ch[p][c]=np,p=fail[p];
        if(!p) fail[np]=1;
        else
        {
            int q=ch[p][c];
            if(ml[q]==ml[p]+1) fail[np]=q;
            else
            {
                int nq=++tot;
                fail[nq]=fail[q];
                for(int i=0; i<26; i++) ch[nq][i]=ch[q][i];
                ml[nq]=ml[p]+1;
                fail[q]=fail[np]=nq;
                while(p&&ch[p][c]==q) ch[p][c]=nq,p=fail[p];
            }
        }
    }
    void calc()
    {
        for(int i=1; i<=tot; i++) c[ml[i]]++;
        for(int i=1; i<=tot; i++) c[i]+=c[i-1];
        for(int i=1; i<=tot; i++) a[c[ml[i]]--]=i;
        for(int i=tot; i>=1; i--)
        {
            int u=a[i];
            siz[fail[u]]+=siz[u];
            ans+=(ll)(ml[u]-ml[fail[u]])*(n-siz[u])*siz[u];
        }
    }
}sam;

int main()
{
    scanf("%s",st+1); n=strlen(st+1);
    sam.las=sam.tot=1;
    for(int i=1; i<=n; i++) sam.ins(st[i]-'a');
    sam.calc();
    printf("%lld",ans);
}