算法训练第四十二天 | 动态规划专题01背包问题、LeetCode 416

01背包实现代码

public class _01背包问题 {
    public static void main(String[] args) {
        int[] weight = {1, 3, 4};
        int[] value = {15, 20, 30};
        int bagSize = 4;
        System.out.println("二维数组");
        bagProblem2(weight, value, bagSize);
        System.out.println("一维数组");
        bagProblem1(weight,value,bagSize);
    }
    //二维背包
    public static void bagProblem2(int []weight,int[] value,int bagSize){
        int wLen = weight.length, value0 = 0;
        //定义dp数组：dp[i][j]表示背包容量为j时，前i个物品能获得的最大价值
        int[][] dp = new int[wLen + 1][bagSize + 1];
        //对于背包问题的初始化：当容量为0的时候，最大价值也一定是0，没办法放物品
        for (int i = 0; i <= wLen; i++) {
            dp[i][0] = value0;
        }
        //而对于后续的遍历，是先遍历物品，再遍历背包
        //即当选第i个物品的时候，根据背包容量不同来确定最大价值
        for (int i = 1; i <= wLen; i++) {
            for (int j = 1; j <= bagSize; j++) {
                //若当前背包容量小于物品重量，即无法放入物品，那么最大价值和前面的一样
                if(j < weight[i-1]){
                    //不放入物品i（可以看成舍掉i），那么剩下的物品就是i-1,个，背包容量还是j
                    dp[i][j] = dp[i-1][j];
                }else{
                    //发现背包容量可以放入物品i，那么这里有两种选择
                    //选择①：不放入物品i（可以看成舍掉i），那么剩下的物品就是i-1,个，背包容量还是j
                    //选择②：放入物品i，剩下的物品就是i-1个，背包容量是，j-放入i的容量，最终价值是加上i的价值
                    dp[i][j] = Math.max(dp[i-1][j],dp[i-1][j-weight[i-1]] + value[i-1]);
                }
            }
        }
        //打印dp数组
        for (int i = 0; i <= wLen; i++){
            for (int j = 0; j <= bagSize; j++){
                System.out.print(dp[i][j] + " ");
            }
            System.out.print("\n");
        }
    }
    public static void bagProblem1(int []weight,int[] value,int bagSize){
        //一维数组
        //定义dp数组：dp[j]表示背包容量为j时，能获得的最大价值
        int []dp = new int[bagSize+1];
        //初始化：容量为0，自然最大价值为0
        dp[0] = 0;
        //逆序遍历保证每个物品只被添加一次
        for (int i = 0; i < weight.length; i++) {
            for (int j = bagSize; j >= weight[i]; j--) {
                dp[j] = Math.max(dp[j],dp[j-weight[i]]+value[i]);
            }
        }
        //打印dp数组
        for (int j = 0; j <= bagSize; j++){
            System.out.print(dp[j] + " ");
        }
    }
}

---

LeetCode 416分割等和子集

题目简析：

给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。

思路分析：见注释

    public boolean canPartition(int[] nums) {
        //01背包问题----每个元素只能选一次
        //因为求的是分割成两个子集，左右的总和相等，即先求出总和，若为奇数则无法分割
        if(nums == null || nums.length == 0) return false;
        int len = nums.length;
        int sum = 0;
        for(int num : nums){
            sum += num;
        }
        //总和为奇数，不能平分
        if(sum % 2 != 0) return false;
        int tar = sum/2;
        //首先明确dp的含义
        //dp[j]表示 背包总容量（所能装的总重量）是j
        //  即放进物品后，背的最大重量为dp[j]
        //那么这个定义就决定了我们的最终求解
        //因为分割（左右各为总和一半）
        //    即说明，容量为j的背包的最大重量若为sum/2，那么就说明可以装，即可达总和一半
        int []dp = new int[tar+1];
        for (int i = 0; i < len; i++) {
            for (int j = tar; j >= nums[i]; j--) {
                //有两种状态：不选num[i]，那么背包容量还是j，价值不变
                //          选num[i]，那么背包容量剩j-num[i]（重量）
                //              背包价值加上num[i]的价值
                dp[j] = Math.max(dp[j],dp[j-nums[i]]+nums[i]);
            }
        }
        return dp[tar] == tar;
    }