HDU 2639 - Bone Collector II (第K优解)

Bone Collector II

题目链接：HDU 2639 Bone Collector II

 

本题要求的是第K优解，我们在01背包中，每次求得的是当前下的最优解，也就是说max（）每次都取了最好的那个解，这里要求的是第K优解，我们改变一下max这里的求解过程，改为求K个解，从最优解一直到第K优解。用a[l]表示在背包容量为l的情况下，当前物品i（前i件物品）不放进背包，背包的价值。用b[l]表示在背包容量为l的情况下，当前物品i（前i件物品）放进背包，背包的价值。在当前物品的情况下，总共有2^k种可能，也就是说每个第k解总会有两种取值选择（放i物品，不放i物品）。

因为题目中说了we considerate two ways that get the same value of bones are the same。所以当出现重复的解的时候，我们按一种情况算。这样就是下面代码的作用：

if(dp[l][z] != dp[l][z-1])
    z++;

AC代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define INF 0x3f3f3f3f
#define N 100001

using namespace std;

int main()
{
    int n, m, k, t;
    int dp[1001][31];
    int w[100], v[100];
    int a[32], b[32];
    scanf("%d", &t);
    for(int j=0; j<t; j++) {
        scanf("%d%d%d", &n, &m, &k);
        memset(dp, 0, sizeof(dp));
        for(int i=0; i<n; i++)
            scanf("%d", &v[i]);
        for(int i=0; i<n; i++)
            scanf("%d", &w[i]);
        for(int i=0; i<n; i++) {
            for(int l=m; l>=w[i]; l--) {
                for(int p=1; p<=k; p++) {
                    a[p] = dp[l][p];
                    b[p] = dp[l-w[i]][p] + v[i];
                }
                a[k+1] = b[k+1] = -1;
                int x=1, y=1, z=1;
                while(z<=k && (x<=k || y<=k)) {
                    if(a[x] > b[y])
                        dp[l][z] = a[x++];
                    else dp[l][z] = b[y++];
                    if(dp[l][z] != dp[l][z-1])
                        z++;
                }
            }
        }
        printf("%d\n", dp[m][k]);
    }
    
    
    return 0;
}