A. chino with string(ac自动机+floyd矩阵快速幂)

issue是fw

于 2021-07-15 23:15:19 发布

阅读量530

点赞数

分类专栏：字符串算法 gym 文章标签： AC自动机矩阵快速幂字符串组合最大价值动态规划

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

版权

字符串算法同时被 2 个专栏收录

92 篇文章 2 订阅

订阅专栏

gym

70 篇文章 0 订阅

订阅专栏

LINK

有 $m$ 个字符串,每个字符串有一定的分值(可能为负数)

求出一个长 $n$ 的字符串 $s$ 使得它的价值最大,你只需要输出这个最大的价值.

价值定义为 $\sum\limits_{i=1}^m ci_i*point_i$

其中 $ci_i$ 表示给定的第 $i$ 个字符串在 $s$ 中的出现次数, $point_i$ 表示第 $i$ 个字符串的分值

对 $m$ 个字符串建 $a c$ 自动机,每个节点预处理一个 $point_i$ 表示以这个节点结尾的所有串的分值和

当然这个 $p o i n t$ 要在 $f a i l$ 树上推标记累加

容易想到定义 $f [i] [j]$ 表示长度为 $i$ 的字符串在 $j$ 节点的最大分值

转移方程一目了然(满足 $q$ 节点到 $j$ 节点有边)

$f[i][j]=\max_{q->j}\{f[i-1][q]+point_j\}$

然而 $n<=10^9$ ,一般来说可以使用矩阵快速幂,但这里是 $\max$ 运算

初始矩阵为,长度是自动机上的节点个数,也就是相当于 $f [0] [0], f [0] [1] . . . f [0] [i d]$ 的值

$\begin{bmatrix} 0 & -inf & \cdots & -inf \\ \end{bmatrix}$

转移矩阵为 $i d * i d$ 的矩阵(下面用 $n$ 表示 $i d$ )

$\begin{bmatrix} x_{11} & x_{12} & \cdots & x_{1n} \\ x_{21} & x_{22} & \cdots & x_{2n} \\ \vdots & \vdots & \ddots & \vdots \\ x_{n1} & x_{n2} & \cdots\ & x_{nn} \\ \end{bmatrix}$

其中 $x_{i,j}$ 表示,若 $i$ 节点到 $j$ 节点有边, $x_{i,j}=mx_j$

若 $i$ 节点到 $j$ 节点无边, $x_{i,j}=-inf$ 代表不能转移

这是一个外层 $\max$ 运算,内层 $+$ 运算的矩阵乘法

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int maxn = 1e6+10;
const ll inf = 1e17;
int n,m,mx;
struct rce
{
	ll m[201][201];
	rce(){ memset( m,-0x3f3f3f3f,sizeof m); }
};
rce operator * ( rce a, rce b )
{
	rce ans;
	for(int i=0;i<=mx;i++)
	for(int j=0;j<=mx;j++)
	for(int k=0;k<=mx;k++)
		ans.m[i][j] = max( ans.m[i][j],a.m[i][k]+b.m[k][j] );
	return ans;
}
int zi[maxn][30],fail[maxn],id = 1;
ll point[maxn];
char a[maxn];
void insert(char a[],int val)
{
	int len = strlen( a+1 ), now = 0;
	for(int i=1;i<=len;i++)
	{
		if( !zi[now][a[i]-'a'] )	zi[now][a[i]-'a'] = ++id;
		now = zi[now][a[i]-'a'];
	}
	point[now] += val;
}
void make_fail()
{
	queue<int>q;
	for(int i=0;i<=25;i++)
		if( zi[0][i] )	q.push( zi[0][i] );
	while( !q.empty() )
	{
		int u = q.front(); q.pop();
		point[u] += point[fail[u]];
		for(int i=0;i<=25;i++)
		{
			int v = zi[u][i];
			if( v )
				fail[v] = zi[fail[u]][i], q.push( v );
			else	zi[u][i] = zi[fail[u]][i];
		}
	}
}
rce quick(rce x,int n)
{
	rce ans = x;
	for( ; n ; n>>=1,x=x*x )
		if( n&1 )	ans = ans*x;
	return ans;	
}
int main()
{
	cin >> n >> m;
	for(int i=1;i<=m;i++)
	{
		int val;
		scanf("%s%d",a+1,&val );
		insert( a,val );
	}
	make_fail(); mx = id;
	rce z;
	
	for(int i=0;i<=id;i++)
	for(int j=0;j<=25;j++)
	{
		int v = zi[i][j];
		z.m[i][v] = point[v];
	}
	z = quick( z,n-1 );			
	ll ans = -inf;
	for(int i=0;i<=mx;i++)	ans = max( ans,z.m[0][i] );
	cout << ans;
}