HDU 2639 (01背包第k优解)

时间:2022-12-02 18:05:56
/*
01背包第k优解问题
f[i][j][k] 前i个物品体积为j的第k优解
对于每次的ij状态 记下之前的两种状态 i-1 j-w[i] (选i) i-1 j (不选i) 分别k个
然后归并排序并且去重生成ij状态的前k优解
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 1010
using namespace std;
int T,n,m,k,c[maxn],v[maxn],f[maxn][],x[maxn],y[maxn],a,b,z;
int init()
{
int x=;bool f=;char s=getchar();
while(s<''||s>''){if(s=='-')f=;s=getchar();}
while(s>=''&&s<=''){x=x*+s-'';s=getchar();}
if(f)return -x; return x;
}
int main()
{
T=init();
while(T--)
{
memset(f,,sizeof(f));
n=init();m=init();k=init();
for(int i=;i<=n;i++)
v[i]=init();
for(int i=;i<=n;i++)
c[i]=init();
for(int i=;i<=n;i++)
{
for(int j=m;j>=c[i];j--)
{
for(int l=;l<=k;l++)
{
x[l]=f[j][l];
y[l]=f[j-c[i]][l]+v[i];
}
a=b=z=;
x[k+]=y[k+]=-;
while(z<=k&&(x[a]!=-||y[b]!=-))
{
if(x[a]<y[b])f[j][z]=y[b],b++;
else f[j][z]=x[a],a++;
if(f[j][z]!=f[j][z-])z++;
}
}
}
printf("%d\n",f[m][k]);
}
return ;
}