BZOJ3238 后缀自动机+推公式

拿到题目第一步就把前面的len（Ti）+len（Tj）手算出来了

可以知道这个答案只和n有关系 为n*(n+1)*(n-1)/2 （如果不会手算On跑出来也行 问题不大）

后面求LCP就是重点了 通常来讲 一看到LCP我就准备放弃了

只因为我在题目之中多看了它一眼 于是我就发现这题可做

首先我们需要一颗后缀树 因为两个后缀的LCP就是它们LCA的len

后缀树怎么得到呢 直接把字符串反向建自动机就行了（基础知识点）

于是求LCA的len就是经典的操作了 求出来以后假设当前状态有n个点 那么可能的组合就有C(n,2)种

刚好题目要乘以2 那么最后答案就是len*n*(n-1) 于是这题就做出来了

不得不说后缀自动机真的太强大了 如此复杂的问题 如果不用自动机 代码会非常非常复杂 而套上自动机就能秒杀

这几天没白熬夜QAQ

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e5+5;
char s[maxn];
int n;
long long ans;
struct Suffix_Automation
{
    int next[maxn*2][26],fa[maxn*2],l[maxn*2],right[maxn*2];
    int last,cnt;
    int cntA[maxn*2],A[maxn*2];
    void init()
    {
        last=cnt=1;
        fa[1]=l[1]=0;
        memset(next[1],0,sizeof next[1]);
    }
    int inline newnode()
    {
        cnt++;
        memset(next[cnt],0,sizeof next[cnt]);
        fa[cnt]=l[cnt]=0;
        return cnt;
    }
    void add(int c)
    {
        int p=last;
        int np=newnode();
        l[np]=l[p]+1;
        last=np;
        while(p&&!next[p][c])
        {
            next[p][c]=np,p=fa[p];
        }
        if(!p)
            fa[np]=1;
        else
        {
            int q=next[p][c];
            if(l[q]==l[p]+1)
                fa[np]=q;
            else
            {
                int nq=++cnt;
                l[nq]=l[p]+1;
                memcpy(next[nq],next[q],sizeof next[q]);
                fa[nq]=fa[q];
                fa[np]=fa[q]=nq;
                while(next[p][c]==q)
                    next[p][c]=nq,p=fa[p];
            }
        }
    }
    void work()
    {
        memset(cntA,0,sizeof cntA);
        for(int i=1;i<=cnt;i++)
            cntA[l[i]]++;
        for(int i=1;i<=n;i++)
            cntA[i]+=cntA[i-1];
        for(int i=1;i<=cnt;i++)
            A[cntA[l[i]]--]=i;
        int p=1;
        for(int i=n;i>=1;i--)
        {
            p=next[p][s[i]-'a'];
            right[p]=1;
        }
        for(int i=cnt;i>=1;i--)
        {
            p=A[i];
            if(fa[p])
                right[fa[p]]+=right[p];
        }
    }
    void solve()
    {
        for(int i=1;i<=cnt;i++)
        {
            ans-=1ll*(l[i]-l[fa[i]])*right[i]*(right[i]-1);
        }
        printf("%lld\n",ans);
    }

}sam;
int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    ans=1ll*n*(n+1)/2*(n-1);
    sam.init();
    for(int i=n;i>=1;i--)
    {
        sam.add(s[i]-'a');
    }
    sam.work();
    sam.solve();
    return 0;
}