loj2377. 「AHOI2013」 差异

　

2377. 「AHOI2013」 差异

给定一个长度为 $n$ 的字符串 $S$ ，令 $T_i$ 表示它从第 $i$ 个字符开始的后缀，求：
$$\sum_{1 \le i < j \le n} \operatorname{len}(T_i)+\operatorname{len}(T_j)-2\operatorname{lcp}(T_i,T_j)$$

---

sol

建出反串，计算每个点会被当几次lcp（有几对特殊点经过）

我居然忘记ch[i]-'a'了气死

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 1000005
using namespace std;
int n,rt,cnt,la,sz[maxn];
long long val[maxn],ans;
char ch[maxn];
struct node{
    int par,nex[26],Max;
}s[maxn];
void ins(int c){
    int p=la,np=++cnt;la=np;s[np].Max=s[p].Max+1;sz[np]=1;
    for(;p&&!s[p].nex[c];p=s[p].par)s[p].nex[c]=np;
    if(!p)s[np].par=rt;
    else {
        int q=s[p].nex[c],nq;
        if(s[q].Max==s[p].Max+1)s[np].par=q;
        else {
            nq=++cnt;s[nq].Max=s[p].Max+1;
            for(int j=0;j<26;j++)s[nq].nex[j]=s[q].nex[j];
            s[nq].par=s[q].par;s[q].par=s[np].par=nq;
            for(;p&&s[p].nex[c]==q;p=s[p].par)s[p].nex[c]=nq;
        }
    }
}
int tax[maxn],ord[maxn];
void Sort(){
    for(int i=1;i<=cnt;i++)tax[s[i].Max]++;
    for(int i=1;i<=n;i++)tax[i]+=tax[i-1];
    for(int i=1;i<=cnt;i++)ord[tax[s[i].Max]--]=i;
}
int main(){
    scanf(" %s",ch);n=strlen(ch);
    la=rt=cnt=1;
    for(int i=n-1;i>=0;i--)ins(ch[i]-'a');
    Sort();
    for(int i=cnt;i>=1;i--){
        int x=ord[i];
        val[s[x].par]+=1LL*sz[s[x].par]*sz[x];
        sz[s[x].par]+=sz[x];
    }
    for(int i=1;i<=n;i++)ans=ans+1LL*i*(n-1);
    for(int i=1;i<=cnt;i++)ans=ans-val[i]*s[i].Max*2;
    cout<<ans<<endl;
    return 0;
}