[后缀自动机 构建后缀树 树形DP] BZOJ 3238 [Ahoi2013]差异

反串的parent树就是后缀树
lcp就是他们的lca的深度 也就是后缀自动机上的maxl
然后树形DP一下就好了

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
  return *p1++;
}

inline int read(char *s){
  char c=nc(); int len=0;
  for (;!(c>='a' && c<='z');c=nc()) if (c==EOF) return 0;
  for (;c>='a' && c<='z';s[++len]=c,c=nc()); s[++len]=0; return len-1;
}

const int N=500005;

struct state{
  int len,link,next[26];
}st[N<<1];
int ncnt,last;

int cnt[N<<1];
ll f[N<<1],Ans;

inline void Extend(char c){
  int cur=++ncnt,p; c-='a';
  st[cur].len=st[last].len+1; cnt[cur]=1; f[cur]=1;
  for (p=last;p!=-1 && !st[p].next[c];p=st[p].link)
    st[p].next[c]=cur;
  if (p==-1)
    st[cur].link=0;
  else{
    int q=st[p].next[c];
    if (st[q].len==st[p].len+1)
      st[cur].link=q;
    else{
      int nq=++ncnt;
      st[nq].len=st[p].len+1;
      st[nq].link=st[q].link;
      for (int i=0;i<26;i++) st[nq].next[i]=st[q].next[i];
      for (;p!=-1 && st[p].next[c]==q;p=st[p].link)
    st[p].next[c]=nq;
      st[q].link=st[cur].link=nq;
    }
  }
  last=cur;
}

int tmp[N<<1],sum[N<<1];

inline void Sort(){
  for (int i=0;i<=ncnt;i++) sum[st[i].len]++;
  for (int i=1;i<=st[last].len;i++) sum[i]+=sum[i-1];
  for (int i=ncnt;~i;i--) tmp[sum[st[i].len]--]=i;
}

char A[N];
int n;


int main(){
  freopen("t.in","r",stdin);
  freopen("t.out","w",stdout);
  n=read(A);
  st[0].link=-1;
  for (int i=n;i;i--) Extend(A[i]);
  Sort();
  for (int i=ncnt+1;i>1;i--) cnt[st[tmp[i]].link]+=cnt[tmp[i]];
  for (int i=ncnt+1;i>1;i--){  
    int x=tmp[i];  
    Ans+=(ll)f[st[x].link]*cnt[x]*st[st[x].link].len;  
    f[st[x].link]+=cnt[x];  
  }
  printf("%lld\n",(ll)(n+1)*n/2*(n-1)-Ans*2);  
  return 0;
}