题意:容量为V的背包,n个物品,每个物品有一个费用及价值,问背包装这些东西所能得到的最K大价值。
分析:01背包求第K优解。
Code:
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=1010;
int f[maxn][33],val[110],cost[110];
int ori[33],now[33];
int main()
{
int T,n,V,k;
scanf("%d",&T);
while(T--){
scanf("%d %d %d",&n,&V,&k);
for(int i=1;i<=n;i++)
scanf("%d",&val[i]);
for(int i=1;i<=n;i++)
scanf("%d",&cost[i]);
memset(f,0,sizeof(f));
for(int i=1;i<=n;i++){
for(int j=V;j>=cost[i];j--){
for(int s=1;s<=k;s++){
ori[s]=f[j][s];
now[s]=f[j-cost[i]][s]+val[i];
}
ori[k+1]=now[k+1]=-1;
int idxo=1,idxn=1,idxk=1;
while(idxk<=k&&(ori[idxo]!=-1||now[idxn]!=-1)){
if(ori[idxo]>now[idxn]) f[j][idxk]=ori[idxo++];
else f[j][idxk]=now[idxn++];
if(f[j][idxk]!=f[j][idxk-1]) idxk++;
}
}
}
printf("%d\n",f[V][k]);
}
return 0;
}