字符串（后缀自动机）：Ahoi2013 差异

Description

Input

一行，一个字符串S

Output

一行，一个整数，表示所求值

Sample Input

cacao

Sample Output

54

HINT

2<=N<=500000,S由小写英文字母组成

　　建反向前缀树，O（N）dp求解。

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int N=1000010;
int fa[N],ch[N][26],len[N],cnt,lst;
int cntE,fir[N],to[N],nxt[N],dp[N];
char s[N];
void Init(){lst=cnt=1;}
void Insert(int c){
    int p=lst,np=lst=++cnt;len[np]=len[p]+1;
    while(p&&ch[p][c]==0)ch[p][c]=np,p=fa[p];
    if(!p)fa[np]=1;
    else{
        int q=ch[p][c],nq;
        if(len[p]+1==len[q])
            fa[np]=q;
        else{
            len[nq=++cnt]=len[p]+1;
            fa[nq]=fa[q];fa[q]=fa[np]=nq;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            while(ch[p][c]==q)ch[p][c]=nq,p=fa[p];
        }    
    }
}

void addedge(int a,int b){
    nxt[++cntE]=fir[a];
    to[fir[a]=cntE]=b;
}
int end[N];
long long ans; 
void DFS(int x){
    int sum=0;
    for(int i=fir[x];i;i=nxt[i]){
        DFS(to[i]);
        ans-=2ll*len[x]*dp[to[i]]*dp[x];
        dp[x]+=dp[to[i]];
    }        
}

int main(){
    Init();
    scanf("%s",s+1);
    int l=strlen(s+1);
    ans=1LL*(l-1)*(l+1)*l/2;
    for(int i=l;i>=1;i--)
        Insert(s[i]-'a'),dp[lst]+=1;
    
    for(int i=2;i<=cnt;i++)
        addedge(fa[i],i);
    DFS(1);    
    printf("%lld
",ans);
    return 0;
}