题意:
给出商品价值、体积,以及背包能装的最大容量,求出能装入的第K大的价值。
如能装入的价值为20 12 2,那么第2大的就是12.
思路:
给0-1背包增加一个维度,dpjk代表装入容量为j时第k大的价值。
我们每次合并 dp[j][k]和dp[j-w[i]][k]+v[i],取前k大进行更新。
代码:
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
int dx[10]= {-1,1,0,0};
int dy[10]= {0,0,-1,1};
const int maxn=1000+10;
int v[maxn],w[maxn];
int dp[maxn][50];
int a[maxn],b[maxn];
int main()
{
int t;
cin>>t;
while(t--)
{
memset(dp,0,sizeof(dp));
int N,V,K;
cin>>N>>V>>K;
rep(i,1,N)
{
cin>>v[i];
}
rep(i,1,N)
{
cin>>w[i];
}
rep(i,1,N)
{
dep(j,V,w[i])
{
rep(k,1,K)
{
a[k]=dp[j][k];
b[k]=dp[j-w[i]][k]+v[i];
}
a[K+1]=b[K+1]=-1;
int l=1,r=1,cnt=1;
while((a[l]>=0||b[l]>=0)&&cnt<=K)
{
if(a[l]>b[r])
{
dp[j][cnt]=a[l++];
}
else
{
dp[j][cnt]=b[r++];
}
if(dp[j][cnt]!=dp[j][cnt-1])cnt++;
}
}
}
cout<<dp[V][K]<<endl;
}
}