【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;
}