CodeForces - 580D Kefa and Dishes

Posted on 2022-01-08 16:52  Capterlliar  阅读(26)  评论(0)    收藏  举报

题意:有n道菜,选m道上,每道菜的美味值为a[i],且y跟在x后上可以额外增加美味值z。求最大美味值。

解:只有18个菜,压,都可以压。既然特殊情况只和上一道菜有关,那直接设dp[i][j]为i状态下最后一道菜上j的最大值(唉最大值写着写着忘了取max谁给我个脑子)。由于要上m道菜,中间判一下。

代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define maxn 1005
#define eps 0.000001
#define inf 0x3fffffff
int n,m,k;
ll dp[1<<18][20]={0};
ll a[20];
map<pair<int,int>,ll> mp;
signed main(){
    scanf("%d%d%d",&n,&m,&k);
    for(int i=0;i<n;i++)
        scanf("%lld",&a[i]);
    for(int i=0;i<k;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        mp[make_pair(x-1,y-1)]=z;
    }
    for(int i=0;i<n;i++){
        dp[1<<i][i]=a[i];
    }
    ll ans=0;
    for(int i=1;i<=(1<<n)-1;i++){
        int cnt=0;
        for(int j=0;j<n;j++){
            if((i&(1<<j))==0)
                continue;
            cnt++;
            for(int l=0;l<n;l++){
                if((i&(1<<l)))
                    continue;
                dp[i|(1<<l)][l]=max(dp[i|(1<<l)][l],dp[i][j]+a[l]+mp[make_pair(j,l)]);
            }
        }
        if(cnt==m){
            for(int j=0;j<n;j++){
                if(i&(1<<j))
                    ans=max(ans,dp[i][j]);
            }
        }
    }
    printf("%lld\n",ans);
    return 0;
}
View Code