[Ahoi2013]差异

Description

Input

一行，一个字符串S

Output

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

Sample Input

cacao

Sample Output

54

HINT

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

前面的直接加就可以了。

后面要减去的lcp就是品酒大会的第一问，排完序后用并查集搞一下就可以了。

记得开LL。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<complex>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<vector>
#define maxn 500010
#define LL long long
using namespace std;
char s[maxn];
int n,k,sa[maxn],tmp[maxn],rk[maxn],fa[maxn];
LL size[maxn];
struct data{
  LL k;
  int id;
}lcp[maxn];
bool cmp(int i,int j){
  if(rk[i]!=rk[j]) return rk[i]<rk[j];
  else{
    int ri=i+k<=n?rk[i+k]:-1,rj=j+k<=n?rk[j+k]:-1;
    return ri<rj;
  }
}
void get_sa(){
  for(int i=0;i<=n;i++)
    sa[i]=i,rk[i]=i<n?s[i]:-1;
  for(k=1;k<=n;k*=2){
    sort(sa,sa+n+1,cmp);
    tmp[sa[0]]=0;
    for(int i=1;i<=n;i++)
      tmp[sa[i]]=tmp[sa[i-1]]+(cmp(sa[i-1],sa[i])?1:0);
    for(int i=0;i<=n;i++) rk[i]=tmp[i];
  }
}
void get_lcp(){
  for(int i=0;i<=n;i++) rk[sa[i]]=i;
  int h=0;
  lcp[0].k=0;
  for(int i=0;i<n;i++){
    int j=sa[rk[i]-1];
    if(h>0) h--;
    for(;j+h<n && i+h<n;h++)
      if(s[j+h]!=s[i+h]) break;
    lcp[rk[i]-1].k=h;
  }
}
bool CMP(const data &a,const data &b){
  return a.k>b.k;
}
int find(int x){
  if(fa[x]!=x) fa[x]=find(fa[x]);
  return fa[x];
}
int main(){
  scanf("%s",s);
  n=strlen(s);
  LL ans=0;
  for(int i=0;i<=n;i++) fa[i]=i,size[i]=1;
  get_sa();
  get_lcp();
  for(int i=1;i<n;i++) lcp[i].id=i;
  sort(lcp+1,lcp+n,CMP);
  for(int i=1;i<n;i++){
    int u=find(lcp[i].id),v=find(lcp[i].id+1);
    ans+=lcp[i].k*size[u]*size[v];
    fa[v]=u;
    size[u]+=size[v];
  }
  LL ans1=0;
  for(LL i=n;i>=1;i--)ans1+=i*(i-1)+i*(n-i);
  printf("%lld",ans1-2*ans);
  return 0;
}