leetcode算法基础 动态规划 01背包

01背包模板

  有N件物品和一个最多能背重量为W 的背包。第i件物品的重量是weight[i]，得到的价值是value[i] 。每件物品只能用一次，求解将哪些物品装入背包里物品价值总和最大。

public static void testWeightBagProblem(int[] weight, int[] value, int bagWeight){
    int wLen = weight.length;
    //定义dp数组：dp[j]表示背包容量为j时，能获得的最大价值
    int[] dp = new int[bagWeight + 1];
    //遍历顺序：先遍历物品，再遍历背包容量
    for (int i = 0; i < wLen; i++){
        for (int j = bagWeight; j >= weight[i]; j--){
            dp[j] = Math.max(dp[j], dp[j - weight[i]] + value[i]);
        }
    }

416.分割等和子集

  题目这道题目是要找是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。
  那么只要找到集合里能够出现 sum / 2 的子集总和，就算是可以分割成两个相同元素和子集了。
  一个商品如果可以重复多次放入是完全背包，而只能放入一次是01背包，写法还是不一样的。要明确本题中我们要使用的是01背包，因为元素我们只能用一次。

01背包必要条件

背包的体积为sum / 2

背包要放入的商品（集合里的元素）重量为 元素的数值，价值也为元素的数值

背包如何正好装满，说明找到了总和为 sum / 2 的子集。

背包中每一个元素是不可重复放入。

public boolean canPartition(int[] nums) {
    if(nums == null || nums.length == 0) return false;
    int n = nums.length;
    int sum = 0;
    for(int num : nums){
        sum += num;
    }
    //总和为奇数，不能平分
    if(sum % 2 != 0) return false;
    int target = sum / 2;
    int[] dp = new int[target + 1];
    for(int i = 0; i < n; i++){
        for(int j = target; j >= nums[i]; j--){
            //物品 i 的重量是 nums[i]，其价值也是 nums[i]
            dp[j] = Math.max(dp[j], dp[j-nums[i]] + nums[i]);
        }
    }
    return dp[target] == target;
}

1049. 最后一块石头的重量 II

  题目这道题和上面的思路差不多，也是均分之后的01背包问题。

public int lastStoneWeightII(int[] stones) {
    int sum = 0;
    for (int i : stones) {
        sum += i;
    }
    int target = sum >> 1;
    //初始化dp数组
    int[] dp = new int[target + 1];
    for (int i = 0; i < stones.length; i++) {
        //采用倒序
        for (int j = target; j >= stones[i]; j--) {
            //两种情况，要么放，要么不放
            dp[j] = Math.max(dp[j], dp[j - stones[i]] + stones[i]);
        }
    }
    return sum - 2 * dp[target];
}

背包排列组合问题

  求背包的排列组合数，外循环为物品正序，内循环为背包容量逆序，递推公式dp[j] += dp[j - nums[i]]。

for (int i = 0; i < nums.length; i++) {
    for (int j = size; j >= nums[i]; j--) {
        dp[j] += dp[j - nums[i]];
    }
}

494. 目标和

  题目这道题主要还是思路，如何转化成01背包问题，物品已经确定，该怎样确定背包容量。
  将所有要加的元素当做left组合，所有要减的元素为right组合。那么有 left - (sum - left) = target 得到 left = (target + sum)/2 。此时问题就是在集合nums中找出和为left的组合。

public int findTargetSumWays(int[] nums, int target) {
    int sum = 0;
    for (int i = 0; i < nums.length; i++) sum += nums[i];
    if ((target + sum) % 2 != 0) return 0;
    int size = (target + sum) / 2;
    if(size < 0) size = -size;
    int[] dp = new int[size + 1];
    dp[0] = 1;
    for (int i = 0; i < nums.length; i++) {
        for (int j = size; j >= nums[i]; j--) {
            dp[j] += dp[j - nums[i]];
        }
    }
    return dp[size];
}

474.一和零

  题目本题中strs 数组里的元素就是物品，每个物品都是一个！而m和n相当于是一个背包，两个维度的背包。三重循环来实现两个维度的01背包。

public int findMaxForm(String[] strs, int m, int n) {
    //dp[i][j]表示i个0和j个1时的最大子集
    int[][] dp = new int[m + 1][n + 1];
    int oneNum, zeroNum;
    for (String str : strs) {
        oneNum = 0;
        zeroNum = 0;
        for (char ch : str.toCharArray()) {
            if (ch == '0') {
                zeroNum++;
            } else {
                oneNum++;
            }
        }
        //倒序遍历
        for (int i = m; i >= zeroNum; i--) {
            for (int j = n; j >= oneNum; j--) {
                dp[i][j] = Math.max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
            }
        }
    }
    return dp[m][n];
}