【bzoj4516】[Sdoi2016]生成魔咒 后缀自动机

入坑，持续更新SAM系列题目……

感觉后缀自动机的构建好难理解，背过代码就可以吧

应用好像不难，这道题只需要在所有修改fa的操作时更新ans即可

每个节点对应若干个子串，所有节点代表的所有子串本质不同，节点i代表的子串个数为mx[i]-mx[fa[i]]

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#define maxn 100010

using namespace std;

map<int,int> ch[maxn*2];
int mx[maxn*2],fa[maxn*2];
long long ans;
int n,m,tot=1,last=1;

int calc(int x) {return mx[x]-mx[fa[x]];}

void insert(int x)
{
int p=last,np=last=++tot;
mx[np]=mx[p]+1;
while (p && !ch[p][x]) ch[p][x]=np,p=fa[p];
if (!p) fa[np]=1,ans+=calc(np);
else
{
int q=ch[p][x];
if (mx[p]+1==mx[q]) fa[np]=q,ans+=calc(np);
else
{
int nq=++tot;mx[nq]=mx[p]+1;ch[nq]=ch[q];
fa[nq]=fa[q];ans+=calc(nq)-calc(q);
fa[np]=fa[q]=nq;ans+=calc(np)+calc(q);
while (p && ch[p][x]==q) ch[p][x]=nq,p=fa[p];
}
}
}

int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
insert(x);
printf("%lld\n",ans);
}
return 0;
}