01背包之求第K优解——Bone Collector II

http://acm.hdu.edu.cn/showproblem.php?pid=2639

题目大意是，往背包里赛骨头，求第K优解，在普通01背包的基础上，增加一维空间，那么F[i,v,k]可以理解为前i个物品，放入容量v的背包时，第K优解的值。时间复杂度为O(NVK)。

Talk is cheap.

看代码吧。

import java.util.Scanner;


public class BoneCollector {
    public static void main(String[] sure) {
        int t;
        Scanner sc = new Scanner(System.in);
        t = sc.nextInt();
        while (t-- > 0) {
            int n, v, k;
            n = sc.nextInt();
            v = sc.nextInt();
            k = sc.nextInt();
            int[] val = new int[n + 1];
            int[] vol = new int[n + 1];
            int[][] dp = new int[v + 2][k + 2];
            int[] tp_a = new int[k + 2];
            int[] tp_b = new int[k + 2];

            for (int i = 0; i < n; i++) {
                val[i] = sc.nextInt();
            }
            for (int i = 0; i < n; i++) {
                vol[i] = sc.nextInt();
            }
            for (int i = 0; i < n; i++) {
                for (int j = v; j >= vol[i]; j--) {
                    for (int m = 1; m <= k; m++) {
                        tp_a[m] = dp[j][m];
                        tp_b[m] = dp[j - vol[i]][m] + val[i];
                    }
                    int tmp = 1, a = 1, b = 1;
                    tp_a[k+1] = tp_b[k+1] = -1;
                    //循环，依次将前K优的存到dp数组
                    while (tmp <= k && (a <= k || b <= k)) {
                        if (tp_a[a] > tp_b[b]) {
                            dp[j][tmp] = tp_a[a];
                            a++;
                        } else {
                            dp[j][tmp] = tp_b[b];
                            b++;
                        }
                        if (dp[j][tmp] != dp[j][tmp - 1])
                            tmp++;
                    }
                }
            }
            System.out.println(dp[v][k]);
        }
    }
}