P3346 [ZJOI2015]诸神眷顾的幻想乡(tire树上的广义SAM)

LINK

简化一下题目,意思就是需要把所有路径组成的字符串统计有多少本质不同的串。

---

考虑广义$SAM$,把所有路径串插入进来不久好了吗??

然而复杂度爆炸…

让我们考虑一下,我们$dfs$这棵树

因为这颗树的路径是很多重复的,比如从根节点到$u$的串插入到$SAM$上,节点是$sa{m}_u$

这时$u$有两个儿子$v_1,v_2$,我们直接在$sa{m}_u$后面插入这两个儿子即可

和广义$SAM$构造的过程是一样的

但是这样,只有不转弯的路径被插入进来了…怎么办…

我们把所有叶子节点当作根都$dfs$一遍即可,任何弯曲的路径在某个叶子节点下的$dfs$都会是一条链

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e7+10;
vector<int>vec[maxn];
int val[maxn],n,m,in[maxn];
int zi[maxn][12],fa[maxn],l[maxn],las=1,id=1;
void insert(int p,int c)
{
	if( zi[p][c] )
	{
		int q = zi[p][c];
		if( l[q]==l[p]+1 )	las = q;//完全吻合的节点 
		else
		{
			int nq = ++id; las = nq;
			memcpy( zi[nq],zi[q],sizeof zi[nq] );
			fa[nq] = fa[q], l[nq] = l[p]+1;
			fa[q] = nq;
			for( ; p&&zi[p][c]==q;p=fa[p] )	zi[p][c] = nq;
		}
	}
	else
	{
		int np = ++id; las = np;
		l[np] = l[p]+1;
		for( ; p&&zi[p][c]==0 ; p = fa[p] )	zi[p][c] = np;
		if( !p )	fa[np] = 1;
		else
		{
			int q = zi[p][c];
			if( l[q]==l[p]+1 )	fa[np] = q;
			else
			{
				int nq = ++id;
				memcpy( zi[nq],zi[q],sizeof zi[nq] );
				fa[nq] = fa[q], l[nq] = l[p]+1;
				fa[np] = fa[q] = nq;
				for( ; p&&zi[p][c]==q;p=fa[p] )	zi[p][c] = nq;	
			} 
		}
	}
}
void dfs(int u,int father,int p)
{
	insert( p,val[u] );
	int now = las;
	for( auto v:vec[u] )
	{
		if( v==father )	continue;
		dfs(v,u,now);
	}
}
int main()
{
	cin >> n >> m;
	for(int i=1;i<=n;i++)	scanf("%d",&val[i] );
	for(int i=1;i<n;i++)
	{
		int l,r; scanf("%d%d",&l,&r);
		vec[l].push_back( r );  vec[r].push_back( l );
		in[l]++, in[r]++;
	}
	for(int i=1;i<=n;i++)	if( in[i]==1 )	dfs(i,0,1 );
	long long ans = 0;
	for(int i=1;i<=id;i++)	ans += l[i]-l[fa[i]];
	cout << ans;	
}