题意:
已知物品的个数、背包的容量、每个物品的价值和体积,求第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]);
}
}