HDU 2639 Bone Collector II(0-1揹包第k優解)
阿新 • • 發佈:2018-11-07
題意:
已知物品的個數、揹包的容量、每個物品的價值和體積,求第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]);
}
}