Codeforces Round #321 (Div. 2) D. Kefa and Dishes

题目链接

大意：题意：有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;
}