后缀自动机再复习 + [USACO17DEC] Standing Out from the Herd

here：https://oi-wiki.org/string/sam/

下面转自 KesdiaelKen的雷蒻论坛



来个广义后缀自动机模板题 [USACO17DEC]Standing Out from the Herd

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 100005;
int n, l[MAXN], m, s[MAXN];
long long ans[MAXN];
namespace SAM {
	const int MAXP = 200005;
	const int C = 26;
	int tr[MAXP][C], link[MAXP] = { -1 }, len[MAXP], sz = 1, last;
	int vis[MAXP];
	inline void Insert(int c) {
		int cur = sz++, p;
		len[cur] = len[last] + 1;
		for(p = last; ~p && !tr[p][c]; p = link[p]) tr[p][c] = cur;
		if(!~p) link[cur] = 0;
		else {
			int q = tr[p][c];
			if(len[q] == len[p] + 1) link[cur] = q;
			else {
				int clone = sz++;
				len[clone] = len[p] + 1, link[clone] = link[q];
				memcpy(tr[clone], tr[q], sizeof tr[q]);
				for(; ~p && tr[p][c] == q; p = link[p]) tr[p][c] = clone;
				link[q] = link[cur] = clone;
			}
		}
		last = cur;
	}
	inline void Update(int x, int i) {
		for(; ~x && vis[x] != i && ~vis[x]; x = link[x])
			if(vis[x]) vis[x] = -1;
			else vis[x] = i;
	}
	inline void Solve() {
		m = 0;
		for(int i = 1; i <= n; ++i)
			for(int j = 0, r = 0; j < l[i]; ++j)
				Update(r = tr[r][s[m++]], i);
		for(int i = 1; i < sz; ++i)
			if(~vis[i]) ans[vis[i]] += len[i] - len[link[i]];
		for(int i = 1; i <= n; ++i)
			printf("%lld\n", ans[i]);
	}
}
using namespace SAM;
int main () {
	scanf("%d", &n);
	for(int i = 1; i <= n; ++i) {
		char ch; last = 0;
		while(!isalpha(ch=getchar()));
		while(isalpha(ch))
			++l[i], Insert(s[m++] = ch-'a'), ch = getchar();
	}
	Solve();
}