CF452E Three strings(广义SAM)

最新推荐文章于 2021-07-15 23:15:19 发布

issue是fw

最新推荐文章于 2021-07-15 23:15:19 发布

阅读量287

点赞数

分类专栏：字符串算法

本文链接：https://blog.csdn.net/jziwjxjd/article/details/115173173

版权

字符串算法专栏收录该内容

92 篇文章 2 订阅

订阅专栏

LINK

建立广义 $S A M$

由于三元组一定是相同长度,所以必定在相同的节点内部

所以我们可以枚举每个节点来计算贡献

那么对于一个节点来说,对 $l\in[len[fa_i],len_i]$ 都产生了

$s i z [u] [1] * s i z [u] [2] * s i z [u] [3]$ 的贡献

这样就变成统计 $s i z$ 的问题了

那么有两种写法

①.基数排序循环处理

②.直接在 $p a r e n t$ 树上 $d f s$ 处理

但是!如果广义 $S A M$ 的在线写法直接令 $l a s = 1$ 而不在 $i n s e r t$ 中特判

这个时候基数排序就是错的,必须去 $dfs\ parent$ 树

假的在线广义 $S A M$ : $dfs\ parent$ 树

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 1e9+7;
const int maxn = 8e5+10;
char a[maxn];
int zi[maxn][27],fa[maxn],len[maxn],siz[maxn][4],las = 1, id = 1;
void insert(int c,int index)
{
	int p = las,np = ++id; las = id; 
	siz[np][index] = 1; 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] );
			fa[nq] = fa[q], len[nq] = len[p]+1; 
			fa[np] = fa[q] = nq;
			for( ;p&&zi[p][c]==q;p=fa[p] )	zi[p][c] = nq;
		}
	}
}
int c[maxn],rk[maxn];
ll ans[maxn];
void chiken_sort()
{
	for(int i=1;i<=id;i++)	c[len[i]]++;
	for(int i=1;i<=id;i++)	c[i] += c[i-1];
	for(int i=id;i>=1;i--)	rk[c[len[i]]--] = i;
	for(int i=id;i>=1;i--)
	{
		int u = rk[i];
		int t = 1ll*siz[u][1]*siz[u][2]%mod*siz[u][3]%mod;
		cout << u << " " << siz[u][1] << " " << siz[u][2] << " " << siz[u][3] << endl;
		ans[len[fa[u]]+1] = ( 1ll*ans[len[fa[u]]+1]+t )%mod;
		ans[len[u]+1] = ( 1ll*ans[len[u]+1]-t )%mod; 
		for(int j=1;j<=3;j++)	siz[fa[u]][j] += siz[u][j];
	}
}
vector<int>vec[maxn];
void dfs(int u)
{
	for( auto v:vec[u] )
	{
		dfs(v);
		for(int j=1;j<=3;j++)	siz[u][j] += siz[v][j];
	}
	cout << u << " " << siz[u][1] << " " << siz[u][2] << " " << siz[u][3] << endl;
	int t = 1ll*siz[u][1]*siz[u][2]%mod*siz[u][3]%mod;
	ans[len[fa[u]]+1] = ( 1ll*ans[len[fa[u]]+1]+t )%mod;
	ans[len[u]+1] = ( 1ll*ans[len[u]+1]-t )%mod; 	
}
void work()
{
	for(int i=2;i<=id;i++)	vec[fa[i]].push_back( i );
	dfs(1);
}
int main()
{
	int w = 1e9;
	for(int i=1;i<=3;i++)
	{
		las = 1;
		scanf("%s",a+1); int le = strlen( a+1 );
		w = min( w,le );
		for(int j=1;j<=le;j++)	insert( a[j]-'a',i );
	}
	chiken_sort();
	work();
	for(int i=1;i<=w;i++)
	{
		ans[i] = (( 1ll*ans[i-1]+ans[i] )%mod+mod)%mod;
		cout << ans[i] << " ";
	}
}

真的广义 $S A M$ :基数排序

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 1e9+7;
const int maxn = 8e5+10;
char a[maxn];
int zi[maxn][27],fa[maxn],len[maxn],siz[maxn][4],las = 1, id = 1;
void insert(int c,int index)
{
	int p = las;
	if( zi[p][c] ) 
	{
		int q=zi[p][c];
		if( len[q]==len[p]+1 ) las=q;
		else 
		{
			int nq = ++id; las = nq;
			memcpy( zi[nq],zi[q],sizeof zi[nq] );
			len[nq] = len[p]+1;	fa[nq] = fa[q]; fa[q] = nq;
			for(	;p&&zi[p][c]==q;p=fa[p] ) zi[p][c]=nq;
		}
	}
	else
	{ 
		int np = ++id; las = np; 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] );
				fa[nq] = fa[q], len[nq] = len[p]+1; 
				fa[np] = fa[q] = nq;
				for( ;p&&zi[p][c]==q;p=fa[p] )	zi[p][c] = nq;
			}
		}
	}
	siz[las][index]++;
}
int c[maxn],rk[maxn],ans[maxn];
void chiken_sort()
{
	for(int i=1;i<=id;i++)	c[len[i]]++;
	for(int i=1;i<=id;i++)	c[i] += c[i-1];
	for(int i=id;i>=1;i--)	rk[c[len[i]]--] = i;
	for(int i=id;i>=1;i--)
	{
		int u = rk[i], t = 1ll*siz[u][1]*siz[u][2]%mod*siz[u][3]%mod;
		ans[len[fa[u]]+1] = ( 1ll*ans[len[fa[u]]+1]+t )%mod;
		ans[len[u]+1] = ( 1ll*ans[len[u]+1]-t )%mod; 
		for(int j=1;j<=3;j++)	siz[fa[u]][j] += siz[u][j];
	}
}
int main()
{
	int w = 1e9;
	for(int i=1;i<=3;i++)
	{
		las = 1;
		scanf("%s",a+1); int le = strlen( a+1 );
		w = min( w,le );
		for(int j=1;j<=le;j++)	insert( a[j]-'a',i );
	}
	chiken_sort();
	for(int i=1;i<=w;i++)
	{
		ans[i] = (( 1ll*ans[i-1]+ans[i] )%mod+mod)%mod;
		cout << ans[i] << " ";
	}
}