P4081 [USACO17DEC]Standing Out from the Herd P(广义SAM或后缀数组容斥)

LINK

题意

给出$n$个串,求每个字符串拥有多少其他字符串没有的子串(重复子串只计算一次)

---

后缀数组。

用特殊字符拼接起来,那么每个字符串在$SA$中都是连续的若干段

对于每一段$[l,r]$单独求本质不同子串,然后去重只需要减去

$lcp(l-1,l)+lcp(r,r+1)-lcp(l-1,r+1)$

这个应该很好理解,不过$lcp(l-1,r+1)$应该很容易忘记。

用广义$SAM$,很套路的一题

对于每个串,对每个前缀在$SAM$中的节点$k$

从$k$一直往根节点跳$father$染色

如果该节点被本串染过色了,就停下来,如果该节点有大于等于两个颜色,也停下来

最后统计颜色只有一的即可

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+10;
const int N = maxn*41;
int n,zi[maxn][28],fa[maxn],len[maxn],las = 1, id = 1,ans[maxn];
char s[maxn];
int color[maxn],f[maxn];
void insert(int c,int col)
{
	int p = las, np = ++id; las = id;
	len[np] = len[p]+1; 
	for( ; p && zi[p][c]==0 ; p = fa[p] )	zi[p][c] = np;
	if( p==0 )	fa[np] = 1;
	else
	{
		int q = zi[p][c];
		if( len[q]==len[p]+1 )	fa[np] = q;
		else
		{
			int nq = ++id;
			memcpy( zi[nq],zi[q],sizeof zi[nq] );
			len[nq] = len[p]+1; fa[nq] = fa[q]; color[nq] = color[q],f[nq] = f[q];
			fa[np] = fa[q] = nq;
			for( ; p && zi[p][c]==q ;p = fa[p] )	zi[p][c] = nq;
		}
	}
	while( np&&f[np]<=1 )
	{
		if( color[np]==col )	break;//已经被染过色了,那么祖先也一定被染过色 
		f[np]++; color[np] = col;//染色 
		np = fa[np];
	}
}
void build(int col)
{
	las = 1;//重新开始构造 
	scanf("%s",s+1 ); int le = strlen( s+1 );
	for(int i=1;i<=le;i++)	insert( s[i]-'a',col );
}
int main()
{
	cin >> n;
	for(int i=1;i<=n;i++)	build(i);	
	for(int i=2;i<=id;i++)
	{
		if( f[i]!=1 )	continue;
		ans[color[i]] += len[i]-len[fa[i]];
	}
	for(int i=1;i<=n;i++)	cout << ans[i] << endl;
}