616 F. Expensive Strings(广义SAM上简单dp)

Posted 2021-08-12 issue是fw

LINK

建出广义$SAM$后,只需要知道经过每个节点的权值和是多少即可

也就是对每个节点求出$\sum c_i\ast p_{s,i}$

我们很容易知道一个节点内的字符串共出现过$si{z}_u$次,$si{z}_u=\sum p_{u,i}$

不过权值$c_i$似乎不同啊…

没关系,那就每次插入前缀时,令那个节点的权值加上$c_i$

那么这个节点的所有父亲都会有这个收益,最后来一遍树形$dp$即可

不判重的广义$SAM$似乎会挂

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;
const int maxn = 2e6+10;
int n,co[maxn];
int fa[maxn],l[maxn],zi[maxn][30],siz[maxn],id = 1, las = 1;
string a[maxn];
i64 zhi[maxn];
void insert(int c,int col)
{
	int p = las;
	if( zi[p][c] )
	{
		int q = zi[p][c];
		if( l[q]==l[p]+1 )	las = q; 
		else
		{
			int nq = ++id; las = nq;
			l[nq] = l[p]+1, fa[nq] = fa[q];
			memcpy( zi[nq],zi[q],sizeof zi[q] );
			fa[q] = nq;
			for( ; p&&zi[p][c]==q ;p=fa[p] )	zi[p][c] = nq;
		}		
	}
	else
	{
		int np = ++id; las = id;
		l[np] = l[p]+1; siz[np] = 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;
				l[nq] = l[p]+1, fa[nq] = fa[q];
				memcpy( zi[nq],zi[q],sizeof zi[q] );
				fa[np] = fa[q] = nq;
				for( ; p&&zi[p][c]==q ;p=fa[p] )	zi[p][c] = nq;
			}
		}
	}
	zhi[las] += co[col];
}
i64 ans;
vector<int>vec[maxn];
void dfs(int u)
{
	for(auto v:vec[u] )
	{
		dfs( v );
		siz[u] += siz[v];	zhi[u] += zhi[v];
	}
	ans = max( ans,1ll*l[u]*zhi[u] );
}
int main()
{
	cin >> n;
	for(int i=1;i<=n;i++)	cin >> a[i];
	for(int i=1;i<=n;i++)	cin >> co[i];
	for(int i=1;i<=n;i++)
	{
		int le = a[i].length();
		for(int j=0;j<le;j++)	insert( a[i][j]-'a',i );
		las = 1;
	}
	for(int i=2;i<=id;i++)	vec[fa[i]].push_back( i );
	dfs( 1 );
	cout << ans;
}