看完题目还是一脸蒙,百度了半天,看了好几个AC代码才勉强写(凑)出来。
题意很简单,菜单上有n道菜,你可以点m样,每样菜有它自己的幸福感,然后还加入了k个规则,比如在吃了第i样菜之后,再吃第j样菜,可以获得c的幸福感,问最大的幸福感。
1<=m<=m<=18 c<=1e9
因为是DP专题,肯定是用DP来写,但是始终找不到头绪。百度了一下,需要保存两个状态,一是现在所有菜的选择状态,二是选择的最后一道菜是什么菜。特殊规则就用邻接矩阵来存就可以了。
菜的选择状态可以用二进制来保存。2的18次方,也不是很大。令s为所有菜的保存状态,于是可以写出状态转移方程。
写出转移方程就很简单了。一遍下来,就一个错误,在一个位运算的地方忘了加括号了,因为位运算的优先级很低,几乎每用一次位运算,都应该加括号。
//Dp[s|(1<<j)][j]=max(Dp[s|(1<<j)][j],Dp[s][i]+ans[j]+Map[i][j];
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn=20;
const int maxnn=(1<<18)+10;
ll ans[maxn]={0};
ll Map[maxn][maxn]={0};
ll dp[maxnn][maxn]={0};
ll MAX=0;
int m,n,k;
int main()
{
cin>>n>>m>>k;
for(int i=0;i<n;i++)
{
cin>>ans[i];
}
while(k--)
{
int a,b,c;
cin>>a>>b>>c;
Map[a-1][b-1]=c;
}
for(int i=0;i<n;i++)
{
dp[(1<<i)][i]=ans[i];
}
ll tal=(1<<n);
for(int s=0;s<tal;s++)
{
int vvv=0;
for(int i=0;i<n;i++)
{
if((s&(1<<i))!=0)
{
vvv++;
for(int j=0;j<n;j++)
{
if((s&(1<<j))==0)
{
int ss=(s|(1<<j));
dp[ss][j]=max(dp[ss][j],dp[s][i]+ans[j]+Map[i][j]);
}
}
}
}
if(vvv==m)
{
for(int i=0;i<n;i++)
{
if((s&(1<<i))!=0)
{
MAX=max(MAX,dp[s][i]);
}
}
}
}
cout<<MAX<<endl;
}