关于后缀自动机的学习网上有很多,所以我就只说说为什么后缀自动机可以转成后缀树。
个人觉得比较好的后缀自动机的博客:http://blog.sina.com.cn/s/blog_70811e1a01014dkz.html
https://blog.csdn.net/litble/article/details/78997914
图片来源:http://blog.sina.com.cn/s/blog_70811e1a01014dkz.html
当我们构造出了后缀自动机以后(如上图),容易发现每一个节点和他的pre指针指向的节点到根节点之间的字串,他们都有着相同的后缀。而在后缀树当中,我们把所有的后缀压入一个trie树中,为了压缩空间每一条边代表的不仅仅是一个字符,而是一个字符串,同样的,如果两个后缀拥有相同的前缀,就可以公用这段空间。而刚才的后缀自动机中,我们通过pre指针找到了相同的后缀,那么我们将字符串倒过来就好了。而需要注意的是,由于nq指针是虚拟建立的,所以他不能作为后缀节点。
构造出来差不多就是这样啦,绿色节点表示后缀标记(说明走到这里就走完了一个后缀啦),边是一个字符串
dp很好写
#include<cstdio>
#include<cstring>
#include<iostream>
#define LL long long
#define maxn 500020*2
using namespace std;
int son[maxn][26],fail[maxn],size[maxn],head[maxn],len[maxn],cnt=1,last=1,tot=1,rt=1;
char s[maxn];LL ans;
struct edge{int v,next;}e[maxn*4];
void adde(int a,int b){e[cnt].v=b,e[cnt].next=head[a];head[a]=cnt++;}
void insert(int x){
int p=last,np=++tot,q,nq;
size[np]=1,len[np]=len[p]+1;
while(p&&!son[p][x])son[p][x]=np,p=fail[p];
if(p==0)fail[np]=rt;
else{
int q=son[p][x];
if(len[q]==len[p]+1)fail[np]=q;
else{
nq=++tot;len[nq]=len[p]+1;
memcpy(son[nq],son[q],sizeof(son[q]));
fail[nq]=fail[q];
fail[np]=fail[q]=nq;
while(p&&son[p][x]==q)son[p][x]=nq,p=fail[p];
}
}
last=np;
}
void dfs(int u,int fa){
for(int v,i=head[u];i;i=e[i].next){
dfs(v=e[i].v,u);
size[u]+=size[v];
}
if(u==1)return;
len[u]-=len[fa];
ans-=(LL)len[u]*size[u]*(size[u]-1);
}
int main(){
scanf("%s",s);int l=strlen(s);
for(int i=l-1;i>=0;i--)insert(s[i]-'a');
ans=(LL)l*(l-1)*(l+1)>>1ll;
for(int i=2;i<=tot;i++)adde(fail[i],i);
dfs(1,0);
printf("%lld",ans);
return 0;
}