字符串（后缀自动机）：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=;
 int fa[N],ch[N][],len[N],cnt,lst;
 int cntE,fir[N],to[N],nxt[N],dp[N];
 char s[N];
 void Init(){lst=cnt=;}
 void Insert(int c){
     int p=lst,np=lst=++cnt;len[np]=len[p]+;
     while(p&&ch[p][c]==)ch[p][c]=np,p=fa[p];
     if(!p)fa[np]=;
     else{
         int q=ch[p][c],nq;
         if(len[p]+==len[q])
             fa[np]=q;
         else{
             len[nq=++cnt]=len[p]+;
             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=;
     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+);
     int l=strlen(s+);
     ans=1LL*(l-)*(l+)*l/;
     for(int i=l;i>=;i--)
         Insert(s[i]-'a'),dp[lst]+=;

     for(int i=;i<=cnt;i++)
         addedge(fa[i],i);
     DFS();
     printf("%lld\n",ans);
     return ;
 }