题目链接:点击查看
题目大意:给出 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;
}