牛客2020年七夕节比赛 - F - 牛妹的考验（ AC自动机 + dp ）

题目链接：点击进入

题目

思路

多串匹配，构建AC自动机，对于字符串权值的重复和包含的情况，可以在建 trie 树和求 fail 失配指针的时候进行处理，处理过后对于每个串，也就是每个对应节点，我们会得到它的权值 sum 。然后就是 dp [ i ] [ j ] ，代表长度为 j ，以 i 节点结束的字符串的最大权值。
转移式：
dp [ to ] [ i + 1 ] = max ( dp [ to ] [ i + 1 ] , dp [ j ] [ i ] + sum [ to ] ) ;
（ 当前长度 j , 以 i 结尾，to 代表 i 向下伸展到的点，也就是 i 的子节点）

代码

// Problem: 牛妹的考验
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/7009/F
// Memory Limit: 524288 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define rep(i,x,y) for(register int i = x; i <= y;++i)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=2e3+10;
const int mod=998244353;
const double eps=1e-9;
const int N=5e3+10;
/*--------------------------------------------*/
inline int read()
{
    int k = 0, f = 1 ;
    char c = getchar() ;
    while(!isdigit(c)){if(c == '-') f = -1 ;c = getchar() ;}
    while(isdigit(c)) k = (k << 1) + (k << 3) + c - 48 ,c = getchar() ;
    return k * f ;
}
/*--------------------------------------------*/

int trie[maxn][26];
int fail[maxn];
int tot;
int sum[maxn];
void insert(char *str,int val)
{
	int len=strlen(str),p=0;
	for(int i=0;i<len;i++)
	{
		int to=str[i]-'a';
		if(!trie[p][to])
			trie[p][to]=++tot;
		p=trie[p][to];
	}
	sum[p]+=val;
}
void getfail()
{
	queue<int>q;
	for(int i=0;i<26;i++)
	{
		if(trie[0][i])
		{
			fail[trie[0][i]]=0;
			q.push(trie[0][i]);
		}
	}
	while(q.size())
	{
		int now=q.front();
		q.pop();
		sum[now]+=sum[fail[now]];
		for(int i=0;i<26;i++)
		{
			if(trie[now][i])
			{
				fail[trie[now][i]]=trie[fail[now]][i];
				q.push(trie[now][i]);
			}
			else
				trie[now][i]=trie[fail[now]][i];
		}
	}
}
char s[maxn];
int n,l,x,dp[maxn][1010];

int main() 
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	scanf("%d%d",&n,&l);
	for(int i=1;i<=n;i++)
	{
		scanf("%s %d",s,&x);
		insert(s,x);
	}
	getfail();
	for(int i=0;i<=tot;i++)
		for(int j=0;j<=l;j++)
			dp[i][j]=-inf;
	dp[0][0]=0;
	for(int i=0;i<l;i++)
	{
		for(int j=0;j<=tot;j++)
		{
			for(int k=0;k<26;k++)
			{
				int to=trie[j][k];
				dp[to][i+1]=max(dp[to][i+1],dp[j][i]+sum[to]);
			}
		}
	}
	int ans=-inf;
	for(int i=0;i<=tot;i++)
		ans=max(ans,dp[i][l]);
	printf("%d",ans);
    return 0;
}