大意:题意:有n种菜(每一种菜有一个满意值ai>=0),你准备吃m种,每种一次。但是如果你按某种规则吃两种菜的话会增加额外的满意值,比如规则(xi yi ci)就是你先吃第xi个菜,然后马上吃第yi个菜,那么你就会额外增加ci点满意值。有k个这样的规则,问你吃m种菜后的最大满意值是多少。
解析:一个状态压缩的dp,这里比较特殊的是每种菜都有先后关系,所以我们要记录一下上个菜选的是什么,用二进制按位的思想表示菜是否被选过,这里有一道类似的习题分享给大家取奶酪
下面看代码:
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(int i=a;i<b;i++)
#define rep1(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=20;
ll arr[N],n,m,k;
ll buf[N][N],dp[1<<20][20];//dp[i][j]表示选了(i的二进制1的个数)种菜,最后一个菜是j
ll dfs(ll d,int l,int num)
{
//cout<<d<<endl;
if(dp[d][l]!=-1)
return dp[d][l];//这种状态已经出现过
if(num==m)//已经选了m种菜
return 0;
ll ret =0;
for(int i=1;i<=n;i++)
{
if(d&(1<<i))//已经选了这种菜
continue;
ret=max(ret,dfs(d^(1<<i),i,num+1)+arr[i]+buf[l][i]);//不选这道菜+这道菜的值+附加值
}
return dp[d][l]=ret;
}
int main()
{
ios::sync_with_stdio(false);
cin>>n>>m>>k;
memset(buf,0,sizeof buf);
memset(dp,-1,sizeof dp);
rep(i,1,n+1)
{
cin>>arr[i];
}
rep(i,0,k)
{
int x,y,v;
cin>>x>>y>>v;
buf[x][y]=v;
}
cout<<dfs(0,19,0)<<endl;//因为没有菜和19有关联关系,所以可以从19开始
return 0;
}