洛谷 P4070 [SDOI2016]生成魔咒【SAM】

传送门
其实这道题就是给一个字符串，然后问每一个前缀中有多少个不同的子串。
同样用 \(len\) 数组的性质，如果新加入一个节点 \(x\)，那么它对于整个已加入串新增的不同的子串数量就是 \(len[x]-len[fa[x]]\)。
我试着解释一下为什么，因为 \(fa[x]\) 代表的所有字符串一定就是 \(x\) 的后缀，\(x\) 新增的串长一定大于 \(len[fa[x]]\)，那么 \(x\) 所代表的串中长为 \(len[fa[x]]+1...len[x]\) 的串就是新增的，当然长度唯一嘛，所以新增的数量就是 \(len[x]-len[fa[x]]\)。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=1e5+10;
int last=1,tot=1,n,fa[N*2],len[N*2];
map<int,int> ch[N*2];
LL ans;

int newnode(int id){fa[++tot]=fa[id];len[tot]=len[id];ch[tot]=ch[id];return tot;}
void insert(int c){
	int p=last,np=last=newnode(0);
	len[np]=len[p]+1;
	for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
	if(!p) fa[np]=1;
	else{
		int q=ch[p][c];
		if(len[q]==len[p]+1) fa[np]=q;
		else{
			int nq=newnode(q);len[nq]=len[p]+1;
			fa[q]=fa[np]=nq;
			for(;p&&ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
		}
	}
	ans+=len[np]-len[fa[np]];
	printf("%lld\n",ans);
}

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