算法思想之动态规划(七)——背包问题

作者: 复旦猿 | 来源:发表于2019-06-03 20:44 被阅读0次

    前言

    今天我们继续讨论经典的动态规划问题之背包问题

    背包问题

    问题描述

    一个背包有一定的承重capacity,有n件物品,每件都有自己的价值,记录在数组v中,也都有自己的重量,记录在数组w中,每件物品只能选择要装入背包还是不装入背包,要求在不超过背包承重的前提下,选出物品的总价值最大。

    问题分析

    其实,这类问题和之前讨论的找零钱问题有相似之处。我们使用二维数组dp进行动态规划——dp[i][j]代表了承重为j的情况下放入前i个物品时的最大总价值。当i=0时,即放入第一个物品时,则有:dp[0][j] = (j >= w[0]) ? v[j] : 0;当j=0时,显然有 dp[i][0] = 0。当i>0j>0时,那么,则有如下几种情况:
    (1) 不放第i个物品,总价值为dp[i-1][j]
    (2) 放第i个物品,总价值为dp[i-1][j-w[i]] + v[i],此时,需要j \geq w[i]
    综上,dp[i][j] = max\{ dp[i-1][j],dp[i-1][j-w[j]] +v[i] \}

    代码实现

    通过问题分析,可以很容易得用代码实现,下面给出算法的java实现。

    public class Backpack {
        public int maxValue(int[] w, int[] v, int n, int cap) {
            // write code here
            return core(w, v, n, cap);
        }
    
        public static int core(int[] w, int[] v, int n, int cap) {
            if (n == 0 || w.length == 0 || v.length == 0 || cap == 0) {
                return 0;
            }
    
            int[][] dp = new int[n][cap + 1];
    
            // 初始化第0行
            for (int i = 0; i < cap + 1; i++) {
                dp[0][i] = i >= w[0] ? v[0] : 0;
            }
    
            // 初始化第0列
            for (int i = 0; i < n; i++) {
                dp[i][0] = 0;
            }
    
            for (int i = 1; i < n; i++) {
                for (int j = 1; j < cap + 1; j++) {
                    if (j >= w[i]) {
                        dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - w[i]] + v[i]);
                    } else {
                        dp[i][j] = dp[i - 1][j];
                    }
                }
            }
    
            for (int i = 0; i < n; i++) {
                System.out.println(Arrays.toString(dp[i]));
            }
            return dp[n - 1][cap];
        }
    
        public static void main(String[] args) {
            Backpack backpack = new Backpack();
            int[] w = new int[]{16, 36, 25, 19, 26, 23};
            int[] v = new int[]{619, 363, 582, 163, 487, 344};
            int n = 6;
            int cap = 35;
            int res = backpack.maxValue(w, v, n, cap);
            System.out.println(res);
        }
    }
    

    其他经典问题

    总结

    通过前面几篇博文,我对经典的动态规划问题进行了整理。
    由于本人水平有限,文章难免有欠妥之处,欢迎大家多多批评指正!

    写在最后

    欢迎大家关注我的个人博客复旦猿

    相关文章

      网友评论

        本文标题:算法思想之动态规划(七)——背包问题

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