美文网首页动态规划背包问题
背包问题系列之--01背包

背包问题系列之--01背包

作者: 南湖Giser | 来源:发表于2018-10-10 18:45 被阅读71次

    问题描述:

        小偷深夜潜入一家珠宝店,店里有5件宝物,重量分别为W{1,3,2,4,5},对应的价值为V{200,100,300,150,350 }。小偷随身只携带了一个容量为5的背包,问小偷应如何选择才能使偷得宝物的价值最大?

    解题思路:

        一、建立动态规划数组dp[i][j],表示小偷在背包总容量为j且只有前i件宝物可供选择的情况下能带走的最大价值;
        二、小偷将宝物一件一件的往袋子里装,对于每件物品i:小偷首先需要考虑,携带的容量为j的包有可能装的下这件宝物i吗?换句话说,小偷除了宝物i以外其他宝物都不带走,他的包能不能装下,即j>=w[i]是否成立?
            (1)如果条件j>=w[i]不成立,那么说明这件宝物i小偷无论如何也带不走了,此时他的能带走的最大价值dp[i][j]=dp[i-1][j];
            (2)如果条件j>=w[i]成立,即小偷是有能力带走这件宝物的,但这个狡猾的小偷又要思考了,宝物i我到底要不要拿走呢?如何取舍才能使得我背包的宝物总价值最大呢?
                     i:如果小偷选择不取宝物i,那么此时背包的总价值与只有前i-1件宝物可选择且背包总容量为j时能获得的最大价值是一样的,即dp[i][j]=dp[i-1][j]
                    ii:如果小偷选择宝物i,那么此时背包的总价值为只有前i-1件宝物可选择且背包总容量为j-w[i]时能获得的最大价值再加上宝物i的价值v[i],即dp[i][j]=dp[i-1][j-w[i]]+v[i];
    二者选择总价值最大的即可,接下来我们就按此思路撸代码了......

    Java代码实现:

    public class Main {
        
        public static void main(String[] args) {
            int totalWeight=5;//背包容量
            Treasure[] packages={new Treasure(200, 1),
                                new Treasure(100, 3),
                                new Treasure(300, 2),
                                new Treasure(150, 4),
                                new Treasure(350, 5)};
            System.out.println(solution(packages, totalWeight));
        }
    
        public static int solution(Treasure[] treasures,int bagCapacity){
            int maxValue=-1;
            if(treasures==null||treasures.length==0||bagCapacity<=0){//参数合法性检查
                maxValue=0;
            }else {
                int treasuresNum=treasures.length;
                //定义动态规划数组dp[i][j],记录有前i件物品可供选择且背包容量为j时理科获得的最大价值
                //这里加1是因为考虑了可选择宝物数为0和背包容量为0的情况
                int dp[][]=new int[treasuresNum+1][bagCapacity+1];
                //初始化可选择宝物数为0或者背包容量为0的情况,这个初始化可省略,数组定义时即自动初始化为0了
                for(int i=0;i<treasuresNum;i++){
                    dp[i][0]=0;
                    dp[0][i]=0;
                }
                //开始动态规划,由下而上
                int currentValue=0;
                int currentWeight=0;
                for(int i=1;i<=treasuresNum;i++){
                    currentValue=treasures[i-1].getValue();//注意这里要减1
                    currentWeight=treasures[i-1].getWeight();
                    for(int j=1;j<=bagCapacity;j++){
                        if(currentWeight>j){//当前考虑的宝物重量超过了背包最大容量
                            dp[i][j]=dp[i-1][j];
                        }else {//反之,小偷有能力带走这块宝物,但他需要考虑下是否价值最大
                            dp[i][j]=Math.max(dp[i-1][j], dp[i-1][j-currentWeight]+currentValue);
                        }
                    }
                }
                maxValue=dp[treasuresNum][bagCapacity];
            }
            return maxValue;
        }
    }
    
    class Treasure{
        private int value;//价值
        private int weight;//重量
        public Treasure(int value,int weight) {
            this.setValue(value);
            this.setWeight(weight);
        }
        public int getValue() {
            return value;
        }
        public void setValue(int value) {
            this.value = value;
        }
        public int getWeight() {
            return weight;
        }
        public void setWeight(int weight) {
            this.weight = weight;
        }
    }
    

    复杂度分析

        以上方法的时间和空间复杂度均为O(VN),其中时间复杂度以不可再优化,但空间复杂度还有在优化的空间,优化解法见参见我的另一篇博文背包问题系列之--01背包空间优化解法

    总结

        动态规划现在是编程面试中的热门,如果面试题是求一个问题的最优解(通常是求最大值或者最小值),而且该问题可分解成若干个子问题,并且子问题之间还有重叠的更小子问题,那么我们就可考虑使用动态规划来解决这个问题了。

    相关文章

      网友评论

        本文标题:背包问题系列之--01背包

        本文链接:https://www.haomeiwen.com/subject/ynhmaftx.html