题目大意:
题目的关键在于求
Σlcp(Ti,Tj),(1≤i<j≤n).
用后缀数组求出hei数组,则lcp(Ti,Tj)对应于hei数组的rmq(i+1,j),然后题目转化为求Σrmq(i,j),(2≤i≤j≤n).
然后每个区间hei[i],...,hei[j]一定有一个在最左边的最小值,枚举这个最小值,再二分区间统计即可.
AC code:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
const ll N=500010;
const ll LOGN=20;
ll l,ans;
ll lg2[N],sa[N],t[N],c[N],rk[N],r1[N],r2[N],h[N],hei[N];
ll f[N][LOGN];
char s[N];
void build(){
for(ll i=0;i<l;i++) c[s[i]]=1;
for(ll i=1;i<(1<<10);i++) c[i]+=c[i-1];
for(ll i=0;i<l;i++) rk[i]=c[s[i]];
for(ll i=1,j;rk[sa[l]]!=l;i<<=1){
for(j=0;j<l;j++) r1[j]=j+i<l?rk[j+i]:0;
for(j=0;j<l;j++) c[j]=0;
for(j=0;j<l;j++) c[r1[j]]++;
for(j=1;j<l;j++) c[j]+=c[j-1];
for(j=0;j<l;j++) t[c[r1[j]]--]=j;
for(j=0;j<l;j++) c[j]=0;
for(j=0;j<l;j++) c[rk[j]]++;
for(j=1;j<l;j++) c[j]+=c[j-1];
for(j=l;j>0;j--) sa[c[rk[t[j]]]--]=t[j];
for(r2[sa[1]]=1,j=2;j<=l;j++)
r2[sa[j]]=r2[sa[j-1]]+(rk[sa[j]]!=rk[sa[j-1]]||r1[sa[j]]!=r1[sa[j-1]]);
for(j=0;j<l;j++) rk[j]=r2[j];
}
for(ll i=0,j=0;i<l;h[i]=j,j=j?j-1:0,i++){
for(;rk[i]!=1&&s[sa[rk[i]]+j]==s[sa[rk[i]-1]+j];j++) ;
}
for(ll i=0;i<l;i++) hei[rk[i]]=h[i];
for(ll i=1;i<=l;i++) f[i][0]=hei[i];
for(ll i=1,j;i<=lg2[l];i++){
for(j=1;j<=l;j++){
f[j][i]=f[j][i-1];
if(j+(1<<(i-1))<=l) f[j][i]=min(f[j][i],f[j+(1<<(i-1))][i-1]);
}
}
}
ll rmq(ll f[N][LOGN],ll L,ll R){
return min(f[L][lg2[R-L+1]],f[R-(1<<lg2[R-L+1])+1][lg2[R-L+1]]);
}
int main(){
scanf("%s",s);
l=strlen(s);
for(ll i=1;i<=l;i++) lg2[i]=(ll)log2(i);
for(ll i=l;i>=2;i--) ans+=(i*(i-1)*3)>>1;
build();
for(ll i=2;i<=l;i++){
ll L1=1,R1=i,L2=i,R2=l+1;
while(L1+1!=R1){
ll M=(L1+R1)>>1;
if(rmq(f,M,i-1)>hei[i]) R1=M;
else L1=M;
}
while(L2+1!=R2){
ll M=(L2+R2)>>1;
if(rmq(f,i,M)<hei[i]) R2=M;
else L2=M;
}
ans-=(hei[i]*(i-R1+1)*(L2-i+1))<<1;
}
printf("%lld\n",ans);
return 0;
}