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

   日期:2020-08-28     浏览:76    评论:0    
核心提示:题目链接:点击查看题目大意:给出 n 个模式串每个模式串都有一个价值,要求构造出一个长度为 m 的字符串,每个模式串出现一次都会得到一次相应的价值,问如何构造可以使得最终的价值总和最大题目分析:做过一道加强版的题目:https://blog.csdn.net/qq_45458915/article/details/104235188多余的功能删掉就好,有个不太一样的细节就是,上面的那个题目的字符串不允许共用,但这个题目的字符串允许共用,所以需要在 getfail() 函数中稍作修改,也就是说需要

题目链接:点击查看

题目大意:给出 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;
}

 

 
打赏
 本文转载自:网络 
所有权利归属于原作者,如文章来源标示错误或侵犯了您的权利请联系微信13520258486
更多>最近资讯中心
更多>最新资讯中心
0相关评论

推荐图文
推荐资讯中心
点击排行
最新信息
新手指南
采购商服务
供应商服务
交易安全
关注我们
手机网站:
新浪微博:
微信关注:

13520258486

周一至周五 9:00-18:00
(其他时间联系在线客服)

24小时在线客服