【CodeForces - 580D】Kefa and Dishes (状压dp)
生活随笔
收集整理的這篇文章主要介紹了
【CodeForces - 580D】Kefa and Dishes (状压dp)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題干:
kefa進入了一家餐廳,這家餐廳中有n個菜(0<n≤18),kefa對第i個菜的滿意度為ai(0≤ai≤10^9),并且對于這n個菜有k個規則,如果kefa在吃完第xi個菜之后吃了第yi個菜(保證xi、yi不相等),那么會額外獲得ci(0≤ci≤10^9)的滿意度。kefa要吃m道任意的菜(0<m≤n),但是他希望自己吃菜的順序得到的滿意度最大,請你幫幫kefa吧!?
輸入第一行是三個數:n,m,k?
第二行是n個數,第i個數表示kefa對第i道菜的滿意度為ai?
第三行到第k+2行每行有3個數:xi,yi,ci,表示如果kefa在吃完第xi道菜之后立刻吃了第yi道菜,則會額外獲得ci的滿意度
Examples
Input
2 2 1 1 1 2 1 1Output
3Input
4 3 2 1 2 3 4 2 1 5 3 4 2Output
12Note
樣例1中,按照2,1的順序,可以額外獲得1點滿意度。?
樣例2中,按4,2,1或者2,1,4,都可以額外獲得5點滿意度。
?
解題報告:
? ?dp[i][j],i代表已經吃過了哪些菜,j代表最后吃的是哪個菜,然后更新判斷就好了。
AC代碼:
#include<bits/stdc++.h> #define ll long long using namespace std; ll dp[1<<18][20]; ll a[20],val[20][20]; ll n,m,k; bool ok(ll x) {int res = 0;for(ll i = 1; i<=(1<<n); i<<=1) {if(i & x) res++;}if(res == m) return 1;else return 0; } int main() {ll u,v;scanf("%lld%lld%lld",&n,&m,&k);for(int i = 1; i<=n; i++) {scanf("%lld",a+i);}for(int i = 1; i<=k; i++) {scanf("%lld%lld",&u,&v);scanf("%lld",&val[u][v]);}memset(dp,-1,sizeof dp);for(int i = 1; i<=n; i++) {dp[(1<<i-1)][i] = a[i];}for(int bit = 0; bit<=(1<<n)-1; bit++) {for(int i = 1; i<=n; i++) {if(dp[bit][i] == -1) continue;for(int j = 1; j<=n; j++) {if(i != j && (bit&(1<<(j-1))) ==0) {dp[bit|(1<<(j-1))][j] = max(dp[bit][i] + a[j] + val[i][j],dp[bit|(1<<(j-1))][j]);}}}}ll maxx = 0;for(int bit = 0; bit<=(1<<n)-1; bit++) {if(ok(bit)) {for(int i = 1; i<=n; i++) {maxx = max(maxx,dp[bit][i]);}}}printf("%lld\n",maxx);return 0 ; }總結:還是太菜了、、、、and? 這種東西位運算啥的包裝成函數最好。
總結
以上是生活随笔為你收集整理的【CodeForces - 580D】Kefa and Dishes (状压dp)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 【POJ - 3304 】Segment
- 下一篇: 【CodeForces - 144C】A