BZOJ3238：[AHOI2013]差异——题解

https://www.lydsy.com/JudgeOnline/problem.php?id=3238

https://www.luogu.org/problemnew/show/P4248

参考：https://blog.csdn.net/Vmurder/article/details/42721101

第一道接触后缀树的题，然而不想讲这个东西。

我们只需要知道将串倒着建后缀自动机parent树就是后缀树即可。

然后两个后缀的lcp就是他们的lca的len。

设点u，则过点u的后缀就有su子树的size和个，所以能配出size[u]*(size[u]-1)/2个对，这条路径的长度贡献为(tr[u].l-tr[f].l)

PS：贡献不是tr[u].l，因为过u的后缀最长的不一定为tr[u].l，所以要一段一段处理。

#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
#include<algorithm>
#include<cctype>
using namespace std;
typedef long long ll;
const int N=1e6+;
struct tree{
    int a[],fa,l;
}tr[N];
struct node{
    int to,nxt;
}e[N];
char s[N];
int last,cnt,tot,size[N],head[N];
inline void add(int u,int v){
    e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt;
}
inline void insert(int c){
    int p=last,np=++tot;
    last=np;tr[np].l=tr[p].l+;
    for(;p&&!tr[p].a[c];p=tr[p].fa)tr[p].a[c]=np;
    if(!p)tr[np].fa=;
    else{
    int q=tr[p].a[c];
    if(tr[p].l+==tr[q].l)tr[np].fa=q;
    else{
        int nq=++tot;tr[nq].l=tr[p].l+;
        memcpy(tr[nq].a,tr[q].a,sizeof(tr[q].a));
        tr[nq].fa=tr[q].fa;tr[q].fa=tr[np].fa=nq;
        for(;p&&tr[p].a[c]==q;p=tr[p].fa)tr[p].a[c]=nq;
    }
    }
    size[np]=;
}
ll ans=;
void dfs(int u,int f){
    for(int i=head[u];i;i=e[i].nxt){
    int v=e[i].to;
    dfs(v,u);
    size[u]+=size[v];
    }
    ans-=(ll)size[u]*(size[u]-)*(tr[u].l-tr[f].l);
}
int main(){
    cin>>s+;
    int n=strlen(s+);
    last=tot=;
    for(int i=n;i>=;i--)insert(s[i]-'a');
    for(int i=;i<=tot;i++)add(tr[i].fa,i);
    ans=(ll)(n-)*n*(n+)>>;
    dfs(,);
    printf("%lld\n",ans);
    return ;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++