hdu 2639 第k大01背包

求每个状态里的k优解,然后合并

/*
HDU 2639
求01背包的第k大解。
合并两个有序序列
*/ #include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAXN=;
int dp[][];//dp[i][j]表示容量为i,第j大的值
int value[MAXN];
int weight[MAXN]; int a[];
int b[]; int N,V,K; void DP()
{
memset(dp,,sizeof(dp));
for(int i=;i<N;i++)
for(int j=V;j>=value[i];j--)
{
for(int k=;k<=K;k++)
{
a[k]=dp[j][k];
b[k]=dp[j-value[i]][k]+weight[i];
}
int x,y,z;
x=y=z=;
a[K+]=b[K+]=-;//这个一定要
while(z<=K&&(x<=K||y<=K))//合并两个已经排好序的序列
{
if(a[x]>b[y])dp[j][z]=a[x++];//注意相同的只算一个
else dp[j][z]=b[y++];
if(dp[j][z]!=dp[j][z-])z++;
}
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&N,&V,&K);
for(int i=;i<N;i++)scanf("%d",&weight[i]);
for(int i=;i<N;i++)scanf("%d",&value[i]);
DP();
printf("%d\n",dp[V][K]);
}
return ;
}
上一篇:[BZOJ 3992] [SDOI 2015] 序列统计(DP+原根+NTT)


下一篇:Photoshop打造一枚火星四射的品牌标志