题目:
题解:
后缀自动机的Parent指针反向以后可以形成一个树结构,和AC自动机的Fail树是类似的。这棵树有一个性质:两个子串的最长公共后缀所代表的状态,是这两个子串代表的状态在Parent树上的LCA位置。
原因也很简单,首先顺着Parent指针往上跳达到的就一定是一个串的后缀,并且跳的步数越少后缀越长。那么用最少的步数让两个状态到达一个节点,这个节点肯定就是最长的公共后缀了。
对于这道题目来说,要求的是最长公共前缀,我们可以把字符串倒过来,建立一棵parent树,这样树的最长公共后缀就是原串的最长公共前缀,然后对每个lca加贡献就好了
代码:
#include <cstdio>
#include <cstring>
#define LL long long
using namespace std;
const int N=2000005;
int ch[N][30],fa[N],step[N],p,np,q,nq,last,cnt;
LL ans,size[N],dis[N],val[N];
char st[N];int point[N],nxt[N],v[N],tot,c[N];
void addline(int x,int y,int z)
{
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
void insert(int c)
{
p=last; last=np=++cnt;
step[np]=step[p]+1; size[np]=1;
while (p && !ch[p][c]) ch[p][c]=np,p=fa[p];
if (!p) {fa[np]=1;return;}
q=ch[p][c];
if (step[q]==step[p]+1){fa[np]=q;return;}
nq=++cnt; step[nq]=step[p]+1;
memcpy(ch[nq],ch[q],sizeof(ch[q]));
fa[nq]=fa[q]; fa[q]=fa[np]=nq;
while (ch[p][c]==q) ch[p][c]=nq,p=fa[p];
}
void dfs(int x,int fa)
{
for (int i=point[x];i;i=nxt[i])
if (v[i]!=fa)
{
dis[v[i]]=dis[x]+c[i];
dfs(v[i],x);
val[x]+=(LL)size[x]*size[v[i]]; size[x]+=size[v[i]];
}
}
int main()
{
scanf("%s",st+1);
int l=strlen(st+1);
last=cnt=1;
for (int i=l;i>=1;i--) insert(st[i]-'a');
for (int i=2;i<=cnt;i++)
{
int vv=step[i]-step[fa[i]];
addline(fa[i],i,vv);
}
dfs(1,0);
for (int i=1;i<=cnt;i++) ans+=(LL)dis[i]*val[i];
LL sum=(LL)l*(l+1);
sum=sum/2*(l-1);ans*=2;
printf("%lld",sum-ans);
}