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 ;
}