题解:
后缀自动机用起来非常好用。。
首先根到每个特殊点就是原串的前缀(size=1)
那么我们会发现在parent树上 right集合不断变成后缀
那么他们的lca就是他们的最长公共后缀
于是问题变成了树上给n个点求len[lca]的和
显然dfs一遍就可以完成了
后缀数组做这题也是比较简单的
枚举最小值单调栈维护就可以了
代码:
#include <bits/stdc++.h> #define ll long long #define rint register int #define rep(i,h,t) for (rint i=h;i<=t;i++) #define dep(i,t,h) for (rint i=t;i>=h;i--) using namespace std; const int N=1.1e6; char s[N]; int size[N],len[N],ch[N][26],sum[N]; int lst=1,node=1,t[N],a[N],w[N],fa[N],T,k; void extend(int c) { int f=lst,p=++node; lst=p; len[p]=len[f]+1; w[p]=size[p]=1; while (f&&!ch[f][c]) ch[f][c]=p,f=fa[f]; if (!f) { fa[p]=1; return;}; int x=ch[f][c],y=++node; if (len[f]+1==len[x]) {fa[p]=x; node--;return;}; len[y]=len[f]+1; fa[y]=fa[x]; fa[x]=fa[p]=y; memcpy(ch[y],ch[x],sizeof(ch[x])); while (f&&ch[f][c]==x) ch[f][c]=y,f=fa[f]; } int main() { freopen("1.in","r",stdin); freopen("1.out","w",stdout); ios::sync_with_stdio(false); cin>>s; int n=strlen(s); int l=strlen(s); dep(i,l,1) extend(s[i-1]-'a'); rep(i,1,node) t[len[i]]++; rep(i,1,node) t[i]+=t[i-1]; rep(i,1,node) a[t[len[i]]--]=i; ll ans=0; rep(i,1,n) ans+=1ll*i*(n-1); dep(y,node,1) { int i=a[y]; size[fa[i]]+=size[i]; ans-=2ll*w[fa[i]]*size[i]*len[fa[i]]; w[fa[i]]+=size[i]; } printf("%lld",ans); return 0; }