关闭

01背包问题

标签: 01背包
103人阅读 评论(0) 收藏 举报
分类:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class OneZeroBag {
    public static class Item{
        char name;
        int weight;
        int value;

        public Item(char name, int weight, int value) {
            this.name = name;
            this.weight = weight;
            this.value = value;
        }

        @Override
        public String toString() {
            return "Item{" +
                    "name=" + name +
                    ", weight=" + weight +
                    ", value=" + value +
                    '}';
        }
    }

    public static int get01PackageMaxValue(List<Item> items, int capacity, List<Item> selectedItems){
        int[][] dp = new int[items.size()+1][capacity+1];
        for(int[] dpi : dp) {
            Arrays.fill(dpi, 0);
        }

        boolean[] selected = new boolean[items.size()];
        Arrays.fill(selected, false);

        for(int currCapacity = 1; currCapacity <= capacity; currCapacity++){
            for(int i = 0; i < items.size(); i++){
                int currItemIndex = i + 1;
                Item currItem = items.get(i);
                if(currItem.weight <= currCapacity){
                    dp[currItemIndex][currCapacity] = Math.max(dp[currItemIndex-1][currCapacity-currItem.weight] + currItem.value,
                                                               dp[currItemIndex-1][currCapacity]);

                }else {
                    dp[currItemIndex][currCapacity] = dp[currItemIndex-1][currCapacity];
                }
            }
        }

        System.out.println("二维数组:");
        for(int[] arr : dp){
            System.out.println(Arrays.toString(arr));
        }

       int capacityTemp = capacity;
        for (int k = items.size()-1; k >= 0; k--){
            Item item = items.get(k);
            int index = k+1;
            if(dp[index][capacityTemp] == dp[index-1][capacityTemp-item.weight] + item.value){
                capacityTemp -= item.weight;
                selectedItems.add(item);
            }
        }
        return dp[items.size()][capacity];
    }

    public static void main(String[] args){
        List<Item> items = new ArrayList<>(Arrays.asList(new Item('a', 2, 6),
                                                        new Item('b', 2, 3),
                                                        new Item('c', 6, 5),
                                                        new Item('d', 5, 4),
                                                        new Item('e', 4, 6)));
        List<Item> selectedItems = new ArrayList<>();
        int maxValue = get01PackageMaxValue(items, 10, selectedItems);
        System.out.println("max value:" + maxValue);
        System.out.println("selected items:" + selectedItems);
    }
}

参考资料写的很好:
http://blog.csdn.net/wumuzi520/article/details/7014559
http://blog.csdn.net/mu399/article/details/7722810

0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:9881次
    • 积分:883
    • 等级:
    • 排名:千里之外
    • 原创:81篇
    • 转载:0篇
    • 译文:0篇
    • 评论:0条
    文章分类