码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2015-03-18 00:53:12      阅读:206      评论:0      收藏:0      [点我收藏+]

标签:

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]);
        }
    }
}

 

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

标签:

原文地址:http://www.cnblogs.com/aboutblank/p/4345833.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!