题意:有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; }
浙公网安备 33010602011771号