洛谷 P4248 [AHOI2013]差异 后缀自动机

题目描述

给定一个长度为$n$的字符串$S$，令$T_i$表示它从第$i$个字符开始的后缀。求

$$\sum_{i=1}^{n-1}\sum_{j=i+1}^{n}len(T_i)+len(T_j)-2*lcp(T_i,T_j)$$
其中，$len(a)$表示字符串$a$的长度，$lcp(a,b)$表示字符串$a$和字符串$b$的最长公共前缀。

输入输出格式

输入格式：
一行，一个字符串 $S$ 。

输出格式：
一行，一个整数，表示所求值。

输入输出样例

输入样例#1：
cacao
输出样例#1：
54
说明
对于 100% 的数据，保证 $2⩽n⩽500000$ ，且均为小写字母。

分析：
主要就是求每两个子串的$lcp$。
两个串$lcp$的长度就是$fail$树上的$lca$。
这个东西还没搞懂，所以就不解释了。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#define LL long long

const int maxn=1e6+7;

using namespace std;

int n,cnt,size[maxn],b[maxn],top[maxn],sum[maxn];
char s[maxn];
LL ans;

struct node{
    int len,fail;
    int son[26];
}t[maxn];

void build_sam()
{
    cnt=1;
    int now=1,p,q,clone;
    for (int i=0;i<n;i++)
    {
        int c=s[i]-'a';
        p=now;
        now=++cnt;
        t[now].len=t[p].len+1;
        size[now]=sum[now]=1;
        while (p&&(!t[p].son[c])) t[p].son[c]=now,p=t[p].fail;
        if (!p) t[now].fail=1;
        else
        {
            q=t[p].son[c];
            if (t[p].len+1==t[q].len) t[now].fail=q;
            else
            {               
                clone=++cnt;
                t[clone]=t[q];
                t[clone].len=t[p].len+1;
                t[q].fail=t[now].fail=clone;
                while (p&&(t[p].son[c]==q)) t[p].son[c]=clone,p=t[p].fail;
            }
        }
    }
}

void calc()
{
    for (int i=1;i<=cnt;i++) b[t[i].len]++;
    for (int i=1;i<=n;i++) b[i]+=b[i-1];
    for (int i=1;i<=cnt;i++) top[b[t[i].len]--]=i;  
    for (int i=cnt;i>0;i--) 
      size[t[top[i]].fail]+=size[top[i]];
    ans=0;  
    for (int i=1;i<=n;i++) ans+=(LL)i*(n-1);    
    for (int i=cnt;i>0;i--)
    {
        ans-=(LL)sum[t[i].fail]*size[i]*t[t[i].fail].len*2;
        sum[t[i].fail]+=size[i];
    } 
    printf("%lld",ans);
}

int main()
{
    scanf("%s",s);
    n=strlen(s);
    build_sam();
    calc();
}