后缀自动机 牛客[AHOI2013]差异

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<cmath>
#include<vector>
#include<set>
#include<cstdio>
#include<string>
#include<deque> 
using namespace std;
typedef long long ll;
#define eps 1e-8
#define INF 0x3f3f3f3f
#define maxn 2000005
struct node{
    int len,fa;
    int ch[26];
}trie[maxn];
int last,tot;
ll len;
ll sum,ans;
char str[maxn];
ll size[maxn];
int c[maxn],x[maxn];
void init(){
    last=tot=1;
    memset(size,0,sizeof(size));
    sum=ans=0;
} 
void insert(int c){
    int p=last;
    int np=last=++tot;
    size[tot]=1;
    trie[np].len=trie[p].len+1;
    for(;p&&trie[p].ch[c]==0;p=trie[p].fa) trie[p].ch[c]=np;
    if(p==0) trie[np].fa=1;
    else{
        int q=trie[p].ch[c];
        if(trie[p].len+1==trie[q].len) trie[np].fa=q;
        else{
            int nq=++tot;
            trie[nq]=trie[q];
            trie[nq].len=trie[p].len+1;
            trie[q].fa=trie[np].fa=nq;
            for(;p&&trie[p].ch[c]==q;p=trie[p].fa)
            trie[p].ch[c]=nq;
        }
    }
} 
void cal(){
    //基数排序,按照len值从小到大排序 
    for(int i=0;i<=tot;i++) c[i]=0;
    for(int i=2;i<=tot;i++) c[trie[i].len]++;
    for(int i=2;i<=tot;i++) c[i]+=c[i-1];
    for(int i=tot;i>=2;i--) x[c[trie[i].len]--]=i;
    //从叶子节点开始遍历fail树 
    for(int i=tot;i>=2;i--){
        int c=x[i];
        //(当前节点代表的字符串出现的次数)*(它fail指向的字符串(是前一个字符串的后缀)的长度)*乘以2
        //这里之所以要再乘以size[trie[i].fa]是因为fail指向的那个节点的size可能为0，它可能是我们在构造sam时创建用来将一个点划分为两个点的节点 
        sum+=2*size[c]*(trie[trie[c].fa].len)*size[trie[c].fa];
        //因为trie[c].fa代表的前缀是trie[c]代表前缀的后缀，所以我们要把trie[c]出现的次数叠加到trie[c].fa上面 
        size[trie[c].fa]+=size[c];
    }
}
int main()
{
    scanf("%s",str);
    init();
    //我们需要求的是str所有后缀两两之间的最长公共前缀长度之和，所以我们将str翻转
    //这样问题就转化成为了求翻转之后字符串的所有前缀两两之间的最长公共后缀之和了 
    len=strlen(str);
    for(int i=len-1;i>=0;i--){ 
        insert(str[i]-'a');
    }
    cal();
    ans=len*(len+1)/2*(len-1);
    printf("%lld\n",ans-sum);
    return 0;
}