hdu2639(背包求第k优解)

时间:2023-03-10 01:42:23
hdu2639(背包求第k优解)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2639

题意:给出一行价值,一行体积,让你在v体积的范围内找出第k大的值

分析:dp[i][j][k]表示前i个物品容积为j时的第k优解。那么对于每种状态dp[i][j]都需要维护好前k优解。

每次根据前k优解进行每种取或不取第i件物品,用数组a记录取第i件物品,数组b记录不取,这样

数组a,b了所有能组成j的x种解,最后在x里取前k优解记录下来就好。剩下的肯定不是前k优解了。。。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#define LL long long
#define mod 1000000007
#define inf 0x3f3f3f3f
#define N 10010
using namespace std;
int p[],w[],dp[][],a[],b[];
int main()
{
int t,n,v,k;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&v,&k);
memset(dp,,sizeof(dp));
for(int i=; i<=n; i++)scanf("%d",&p[i]);
for(int i=; i<=n; i++)scanf("%d",&w[i]);
for(int i=; i<=n; i++)
{
for(int j=v; j>=w[i]; j--)
{
for(int l=; l<=k; l++)//组成体积为j无非就是j-w[i]状态加上第i件和不取第i件,最多有2k种
{
a[l]=dp[j-w[i]][l]+p[i];
b[l]=dp[j][l];
}
a[k+]=-;
b[k+]=-;
int x,y,z;
x=y=z=;
while(z<=k&&(a[x]!=-||b[y]!=-))//在最多2k种里取最大的k个解并有序存入
{
if(a[x]>b[y])
{
dp[j][z]=a[x];
x++;
}
else
{
dp[j][z]=b[y];
y++;
}
if(dp[j][z]!=dp[j][z-])z++;
}
}
}
printf("%d\n",dp[v][k]);
}
}