题目
https://www.luogu.com.cn/problem/P4248
思路
考虑建SAM
根据定义,两个点在SAM上的parent树的LCA即为其LCS的距离,由于人以两个子串的LCS=任意两个字串的LCP,所以不用反过来插入(当然反过来也行)
所以我们在SAM上做DP,维护一下子树的size,计算这条边经过多少次
代码
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e6+77;
char st[N];
int n;
ll ans;
struct SAM
{
int tot,las,ch[N][27],fail[N],ml[N],siz[N],c[N],a[N];
void ins(int c)
{
int p=las,np=las=++tot;
ml[np]=ml[p]+1; siz[np]=1;
while(p&&!ch[p][c]) ch[p][c]=np,p=fail[p];
if(!p) fail[np]=1;
else
{
int q=ch[p][c];
if(ml[q]==ml[p]+1) fail[np]=q;
else
{
int nq=++tot;
fail[nq]=fail[q];
for(int i=0; i<26; i++) ch[nq][i]=ch[q][i];
ml[nq]=ml[p]+1;
fail[q]=fail[np]=nq;
while(p&&ch[p][c]==q) ch[p][c]=nq,p=fail[p];
}
}
}
void calc()
{
for(int i=1; i<=tot; i++) c[ml[i]]++;
for(int i=1; i<=tot; i++) c[i]+=c[i-1];
for(int i=1; i<=tot; i++) a[c[ml[i]]--]=i;
for(int i=tot; i>=1; i--)
{
int u=a[i];
siz[fail[u]]+=siz[u];
ans+=(ll)(ml[u]-ml[fail[u]])*(n-siz[u])*siz[u];
}
}
}sam;
int main()
{
scanf("%s",st+1); n=strlen(st+1);
sam.las=sam.tot=1;
for(int i=1; i<=n; i++) sam.ins(st[i]-'a');
sam.calc();
printf("%lld",ans);
}