HDU2639[背包第K大]

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

题意：求第k大背包。

题解：利用二路归并的思想，求解第K大的值。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAXN = ;
int dp[MAXN][], w[MAXN], v[MAXN];
int N, W, K;
int A[], B[];
void super_kth()
{
    for(int i = ; i <= N; i++)
    {
        for(int j = W; j >= w[i]; j--)
        {
            for(int l = ; l <= K; l++)
            {
                A[l] = dp[j - w[i]][l] + v[i]; //放i
                B[l] = dp[j][l];//不放i
            }
            int nu = , x = , y = ;
            A[K + ] = -;
            B[K + ] = -;
            while(nu <= K && (A[x] != - || B[y] != -))
            {
                if(A[x] > B[y])
                    dp[j][nu] = A[x++];
                else
                    dp[j][nu] = B[y++];
                if(dp[j][nu] != dp[j][nu - ])//不相等的第k大背包
                    nu++;
            }
        }
    }
    printf("%d\n", dp[W][K]);
}
int main ()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        memset(dp, , sizeof(dp));
        scanf("%d%d%d", &N, &W, &K);
        for(int i = ; i <= N; i++)
            scanf("%d", &v[i]);
        for(int i = ; i <= N; i++)
            scanf("%d", &w[i]);
        super_kth();
    }
    return ;
}