代码随想录算法训练营第四十三天 | 1049. 最后一块石头的重量 II & 494. 目标和 & 474.一和零

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

1049. 最后一块石头的重量 II - 力扣（LeetCode）

相当于将石头分成尽量相等的两堆，然后总和减去一堆*2，即为剩下的重量

在 0 - i 个石头中装满容量为 sum/2 的背包

石头的重量和价值都是stones[i]

class Solution {
    public int lastStoneWeightII(int[] stones) {
      int sum = 0;
      for(int n: stones){
        sum += n;
      }

      int target = sum >> 1;
      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], Math.max(dp[j-1], dp[j-stones[i]] + stones[i]));
        }
      }

      return sum - dp[target] * 2;
    }
}

2. 目标和

494. 目标和 - 力扣（LeetCode）

left - right = target

left + right = sum

left = (target + sum) / 2

left 不为整，无解

target 的绝对值 大于sum 无解

在 0 - i 个物品 中装满容量为 left 的背包，有多少种方式

dp[j] 表示：填满 j（包括j）这么大容积的包，有dp[j]种方法

确定了一个数（重量）nums[i]，那么就还需要 j - nums[i] 才能装满，即有 dp[j - nums[i]] 种方法。

将nums遍历，取其中一个数确定下来，就能获得对应 这种情况下的方法数量

所有情况的方法总和即为所求 dp[j] += dp[j-nums[i]];

class Solution {
    public int findTargetSumWays(int[] nums, int target) {
        // left - right = target
        // left + right = sum
        // left = (target + sum) / 2

        int sum = 0;
        for(int n: nums){
            sum += n;
        }

        // target 的绝对值 大于sum 无解
        if(target < 0 && sum < -target) return 0;
        // left 不为整，无解
        if((target + sum) % 2 != 0) return 0;

        int size = (target + sum) / 2; // 背包容量
        int[] dp = new int[size + 1];
        //dp[j] 表示：填满 j（包括j）这么大容积的包，有dp[j]种方法

        dp[0] = 1; // +0 -0都是一种方法

        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];

    }
}

3.一和零

474. 一和零 - 力扣（LeetCode）

一维的背包只有一个重量

本题的背包有两个维度 0 和 1 的数量

物品即为每个str， 取它的价值就在于背包内的物品数量可以加1

取或不取 要看两个维度是否满足放进背包的条件（满足了不一定取，还要和不取进行比较）

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        // 背包 有 两个维度 0的数量 1的数量
        int[][] dp = new int[m+1][n+1];
        
        //遍历物品
        for(String str: strs){
            int oneNum = 0;
            int zeroNum = 0;
            for (char ch : str.toCharArray()) {
                if (ch == '0') {
                    zeroNum++;
                } else {
                    oneNum++;
                }
            }

            // 遍历背包
            for(int i = m; i >= zeroNum; i--){ //满足0的维度
                for(int j = n; j >= oneNum; j--){ //满足1的维度
                    dp[i][j] = Math.max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
                }
            }
        }

        return dp[m][n];
    }
}