[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 = ;
 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 = ; i < n; i++) {
             scanf("%d", &v[i]);
         }
         for(int i = ; i < n; i++) {
             scanf("%d", &w[i]);
         }
         memset(dp, , sizeof(dp));
         for(int i = ; i < n; i++) {
             for(int j = m; j >= w[i]; j--) {
                 int cnt = ;
                 for(int p = ; p <= k; p++) {
                     d[cnt++] = dp[p][j];
                     if(j - w[i] >= ) {
                         d[cnt++] = dp[p][j-w[i]] + v[i];
                     }
                 }
                 sort(d, d+cnt, greater<int>());
                 cnt = unique(d, d+cnt) - d;
                 int q = ;
                 for(int p = ; p < cnt; p++) {
                     dp[q++][j] = d[p];
                 }
             }
         }
         printf("%d\n", dp[k][m]);
     }
     return ;
 }