[bzoj3238]差异

对于len(Ti),len(Tj)，随意乱搞就行了。

对于lcp(Ti,Tj)，很显然如果将所有的后缀都搞出来的话比较好弄。

既然是要求lcp的和，那么搞到后缀树上去找节点就行了。

因为后缀树有压缩，所以没有分叉，直接树上dp就行。（即使有分叉也没啥卵用。）

但是如何统计分叉呢？

计数原理。

对于每一个分叉，很显然构成的后缀的数量是在所有结束集合中任选两个，那么就是C²_sz[x]，那么就好搞了，最后乘上一个长度。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <vector>
#include <cmath>
#include <queue>
#include <complex>
#include <set>
using namespace std;

const int N=1500000;

int ch[N][30],len[N],pre[N],r[N],tot=1,last=1;
int to[N],head[N],rest[N],num;
int p,np,q,nq;
long long ans;
char str[N];

void add(int u,int v){
    to[++num]=v;
    rest[num]=head[u];
    head[u]=num;
}

int nd(int l){return len[++tot]=l,tot;}

void add(int v){
    last=np=nd(len[p=last]+1);
    pre[np]=r[np]=1;
    while(p&&!ch[p][v])ch[p][v]=np,p=pre[p];
    if(p){
        pre[np]=q=ch[p][v];
        if(len[p]+1!=len[q]){
            nq=nd(len[p]+1);
            memcpy(ch[nq],ch[q],sizeof ch[q]);
            pre[nq]=pre[q],pre[q]=pre[np]=nq;
            while(p&&ch[p][v]==q)ch[p][v]=nq,p=pre[p];
        }
    }
}

void dp(int x){
    for(int i=head[x];i;i=rest[i]){
        dp(to[i]);
        r[x]+=r[to[i]];
    }
    ans-=(long long)r[x]*(r[x]-1)*(len[x]-len[pre[x]]);
}

int main(){
    scanf("%s",str);
    int len=strlen(str);
    ans=(long long)len*(len-1)*(len+1)/2;
    for(int i=len-1;i>=0;i--)add(str[i]-'a');
    for(int i=2;i<=tot;i++)add(pre[i],i);
    for(int i=head[1];i;i=rest[i])dp(to[i]);
    printf("%lld\n",ans);
}