01背包问题

原创 2016年05月31日 21:59:00
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

版权声明:

相关文章推荐

01背包问题

  • 2012-11-27 13:04
  • 169KB
  • 下载

01背包问题

01背包问题(动态规划入门)

01背包问题 给定N种物品和一个背包。物品i的重量是Wi,其价值位Vi ,背包的容量为M。问应该如何选择装入背包的物品,使得转入背包的物品的总价值为最大?? 在选择物品的时候,对每种物品i只有两...

01背包问题C

  • 2014-05-24 20:07
  • 2.17MB
  • 下载

01背包问题

蓝桥杯——说好的进阶之01背包问题

01背包是在M件物品取出若干件放在空间为W的背包里,每件物品的体积为W1,W2……Wn,与之相对应的价值为P1,P2……Pn

01背包问题

C++编写的01背包问题

  • 2015-07-07 10:52
  • 1.25MB
  • 下载

01背包问题小结

01背包问题

01背包问题 回溯法

内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)