牛客 - 牛妹的考验(AC自动机+dp)

题目链接：点击查看

题目大意：给出 n 个模式串每个模式串都有一个价值，要求构造出一个长度为 m 的字符串，每个模式串出现一次都会得到一次相应的价值，问如何构造可以使得最终的价值总和最大

题目分析：做过一道加强版的题目：https://blog.csdn.net/qq_45458915/article/details/104235188

多余的功能删掉就好，有个不太一样的细节就是，上面的那个题目的字符串不允许共用，但这个题目的字符串允许共用，所以需要在 getfail() 函数中稍作修改，也就是说需要累加一下共用的字符串的贡献，剩下的直接 dp 就好了，dp[ i ][ j ] 代表构造长度为 i 的字符串，当前状态为 j 时的最大贡献，传递的话在AC自动机的trie树上跑就好了

代码：
 

#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
using namespace std;
   
typedef long long LL;
  
typedef unsigned long long ull;
   
const int inf=0x3f3f3f3f;
   
const int N=2e3+100;
 
int n,m;
 
char s[N];
 
int fail[N],trie[N][26],cnt,sum[N];
 
vector<int>vis[N];
 
int dp[1010][N];//dp[i][j]:长度为 i ，状态为 j 时的最大值 
 
void insert_word(int val)
{
	int len=strlen(s);
	int pos=0;
	for(int i=0;i<len;i++)
	{
		int to=s[i]-'a';
		if(!trie[pos][to])
			trie[pos][to]=++cnt;
		pos=trie[pos][to];
	}
	sum[pos]+=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.empty())
	{
		int cur=q.front();
		q.pop();
		sum[cur]+=sum[fail[cur]];
		for(int i=0;i<26;i++)
		{
			if(trie[cur][i])
			{
				fail[trie[cur][i]]=trie[fail[cur]][i];
				q.push(trie[cur][i]);
			}
			else
				trie[cur][i]=trie[fail[cur]][i];
		}
	}
}
 
int main()
{
#ifndef ONLINE_JUDGE
//  freopen("input.txt","r",stdin);
//    freopen("output.txt","w",stdout);
#endif
//  ios::sync_with_stdio(false);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		int val;
		scanf("%s%d",s,&val);
		insert_word(val);
	}
	getfail();
	memset(dp,-inf,sizeof(dp));
	dp[0][0]=0;
	for(int i=0;i<m;i++)
		for(int j=0;j<=cnt;j++)
			for(int k=0;k<26;k++)
			{
				int nj=trie[j][k];
				dp[i+1][nj]=max(dp[i+1][nj],dp[i][j]+sum[nj]);
			}
	int ans=-inf;
	for(int i=0;i<=cnt;i++)
		ans=max(ans,dp[m][i]);
	printf("%d\n",ans);
	
	
	
	
	
 
      
      
      
      
      
      
      
      
      
    return 0;
}