[HDOJ2639]Bone Collector II（第k优01背包）

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

题意：求01背包的第k优解

dp(i, j)表示容量为j时的i优解

对于第二维的操作和01背包几乎一样，只是我们只需要关注每一次取的前k大个即可。

#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <fstream>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>

using namespace std;

const int maxn = 1010;
int n, m, k;
int v[maxn], w[maxn];
int dp[maxn][maxn];
int d[maxn];

int main() {
    freopen("in", "r", stdin);
    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d%d", &n, &m, &k);
        for(int i = 0; i < n; i++) {
            scanf("%d", &v[i]);
        }
        for(int i = 0; i < n; i++) {
            scanf("%d", &w[i]);
        }
        memset(dp, 0, sizeof(dp));
        for(int i = 0; i < n; i++) {
            for(int j = m; j >= w[i]; j--) {
                int cnt = 0;
                for(int p = 1; p <= k; p++) {
                    d[cnt++] = dp[p][j];
                    if(j - w[i] >= 0) {
                        d[cnt++] = dp[p][j-w[i]] + v[i];
                    }
                }
                sort(d, d+cnt, greater<int>());
                cnt = unique(d, d+cnt) - d;
                int q = 1;
                for(int p = 0; p < cnt; p++) {
                    dp[q++][j] = d[p];
                }
            }
        }
        printf("%d
", dp[k][m]);
    }
    return 0;
}