1. 程式人生 > >HDU 2639 Bone Collector II(0-1揹包第k優解)

HDU 2639 Bone Collector II(0-1揹包第k優解)

題意:
已知物品的個數、揹包的容量、每個物品的價值和體積,求第k優解;

思路:
和0-1揹包相似,就是陣列加了多一維,不同的是對於第i個物品選和不選的問題,0-1揹包中是直接求的max(dp[j],d[j-w[i]]+v[i]);而在這裡因為要求第k優解,需要將選(mv[])和不選(mw[])的情況記錄下來(存到陣列中),之後再將兩個
陣列進行有序的合併。


以下AC程式碼:

#include<stdio.h>
#include<string.h>
int dp[1005][35];
int v[1005],w[1005];   
int mv[1005],mw[1005
]; int main(){ int i,j,kk; int n,c,k; int t; scanf("%d",&t); while(t--){ scanf("%d%d%d",&n,&c,&k); for(i=0;i<n;i++) scanf("%d",&v[i]); for(i=0;i<n;i++) scanf("%d",&w[i]); memset(dp,0,sizeof(dp)); for
(i=0;i<n;i++){ for(j=c;j>=w[i];j--){ for(kk=1;kk<=k;kk++){ mv[kk]=dp[j][kk]; mw[kk]=dp[j-w[i]][kk]+v[i]; } mv[k+1]=mw[k+1]=-1; int p=1,q=1,r=1; while(r<=k && (mv[p]!=-1
|| mw[q]!=-1)){ if(mv[p]>mw[q]){ dp[j][r]=mv[p]; p++; } else{ dp[j][r]=mw[q]; q++; } if(dp[j][r]!=dp[j][r-1]) r++; } } } printf("%d\n",dp[c][k]); } }