学习笔记-动态规划解决01背包问题

01背包问题

解决思路

可以通过画一张表来解决。如右图

横轴表示限定最大重量，纵轴表示放入的物品，当指向吉他那一行时表示可以放入吉他，当指向音响那一行时表示能放入吉他和音响，当指向电脑那一行时表示能放入吉他、音响和电脑。表中的数字就表示那个条件下背包的最大价值。最后一个格子就代表最终解。
填表的方式是，首先第一行第一列分别代表背包只有0磅和没有物品，全部填0。然后从G行1磅列开始，如果g的重量大于列限定的重量，显然装不进去，就填入该列上一行的数据，如果g的重量小于等于列限定的重量，说明能放进去，这时如果把g放进去的价值加上剩余容量的最大价值（也可能没有）大于该列上一行的数据，就选择第一种，否则就填入第二种（说明把g加上还不如之前的价值大）。然后依次向下填。
抽象成公式:
w表示物品重量列表，val表示物品价值列表，v表示这个表格，i表示物品，j表示当前背包容量。i，j从1开始
如果 w[i-1]>j v[i][j]=v[i-1][j]
如果 w[i-1]<=j v[i][j]=max(v[i-1][j],val[i-1]+v[i-1][j-w[i-1]])
w[i-1]就表示物品i的重量，因为i从1开始，所以要-1
v[i][j]就表示当前情况下的最大价值
v[i-1][j]就表示同容量下上一种情况的最大价值
j-w[i-1]就表示放入物品i后，背包的剩余容量
v[i-1][j-w[i-1]]就表示上一种情况下，背包剩余容量下的最大价值

填完表格后，只能获得最终价值，为了保存都放入了哪些物品，使用一个与以上表格v同容量的表格path记录放入物品的情况。
执行v[i][j]=v[i-1][j]时，只是在复制上一种情况的最优解，并没有真正放入物品， 而执行v[i][j]=val[i-1]+v[i-1][j-w[i-1]]时，才是真正加入物品的时候。
所以，当执行 v[i][j]=val[i-1]+v[i-1][j-w[i-1]]是，让path[i][j]=1，表示放入该物品。
读取时，因为最后一个格子才是最终解，所以从后往前遍历path，用while循环，i表示最后一个物品，j表示最大容量，如果path[i][j]==1说明放入了新物品，此时输出i，j要等于背包放入i后的剩余容量，这样，下一次循环就应该在限定容量为背包放入i后的剩余容量的列寻找是否放入别的物品，i每次递减。

完整代码

package algorithm;

/**
 * 动态规划，01背包问题
 */
public class KnapsackProblem {
    public static void main(String[] args) {
        int[] w={1,4,3};//重量
        int[] val={1500,3000,2000};//价值
        int m=4;//背包的重量
        int n=val.length;//物品的数量
        int[][] v=new int[n+1][m+1];//表格
        int[][] path=new int[n+1][m+1];//记录放了什么物品
        //按照规则填表
        for (int i = 1; i <v.length ; i++) {
            for (int j = 1; j <v[i].length ; j++) {
                if(w[i-1]>j){
                    v[i][j]=v[i-1][j];
                }else {
                    if(v[i-1][j]>val[i-1]+v[i-1][j-w[i-1]]){
                        //复制之前的最优价值
                        v[i][j]=v[i-1][j];
                    }else {
                        //真正加入物品
                        v[i][j]=val[i-1]+v[i-1][j-w[i-1]];
                        path[i][j]=1;
                    }

                }
            }
        }
        //取出放入的物品
        int i = path.length-1;
        int j = path[i].length-1;
        while (i>0&&j>0){
            if(path[i][j]==1){
                System.out.println("放入第"+i+"个物品");
                j=j-w[i-1];
            }
            i--;
        }
        //输出表格
        for (int x = 0; x <v.length ; x++) {
            for (int y = 0; y <v[x].length ; y++) {
                System.out.print(v[x][y]+" ");
            }
            System.out.println("");
        }
    }
}