代码随想录 动态规划 || 1049 494 474

Day37

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

力扣题目链接

有一堆石头，每块石头的重量都是正整数。

每一回合，从中选出任意两块石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且 x <= y。那么粉碎的可能结果如下：

如果 x == y，那么两块石头都会被完全粉碎；

如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。

最后，最多只会剩下一块石头。返回此石头最小的可能重量。如果没有石头剩下，就返回 0。

思路

• 目标是把石头分为两堆，这两堆重量差尽可能小

• 我们可以遍历数组，得到质量总和sum，向下取整除以2就是每堆石头的目标重量target

• 如果数组中有石头正好可以把这个容量为target的背包填满，那最后一块石头重量就是0或1

• 如果填不满，加入target是10，sum是21，只能找到9，那剩下的石头重量就是21 - 9 - 9 = 3

• 所以问题转化为，有一个容量为target的背包，去装石头，让背包尽量装满（装满就是装的质量尽可能大，石头质量=石头价值，就是装的石头价值尽可能大）

• 所以转换为01背包问题

代码

class Solution {
    public int lastStoneWeightII(int[] stones) {
        int sum = 0;
        for (int stone : stones) {
            sum += stone;//得到石头数组的总质量
        }
        int target = sum >> 1;//目标质量就是sum/2，向下取整
        int[] dp = new int[target + 1];//背包容量为target
        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];//返回最后一块石头的重量
    }
}

494.目标和

力扣题目链接

给定一个非负整数数组，a1, a2, ..., an, 和一个目标数，S。现在你有两个符号 + 和 -。对于数组中的任意一个整数，你都可以从 + 或 -中选择一个符号添加在前面。

返回可以使最终数组和为目标数 S 的所有添加符号的方法数。

思路

• 根据数学推导，可以把问题转化为用一些物品，要装满一个容量为size的背包，共有几种方法

• 确定dp数组以及下标的含义

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

• 确定递推公式

• dp[j] += dp[j - nums[i]]

• dp数组如何初始化

• dp[0] = 1

• 确定遍历顺序

• nums放在外循环，target在内循环，且内循环倒序

代码

class Solution {
    public int findTargetSumWays(int[] nums, int target) {
        int sum = 0;
        for (int num : nums) {
            sum += num;
        }
        if (sum < Math.abs(target)) return 0;//如果target太大，无法达到要求
        if ((sum + target) % 2 != 0) return 0;//如果sum + target为奇数，也没有符合要求的
        int size = (sum + target) >> 1;//背包容量为size
        if (size < 0) size = -size;//保证size大于0
        int[] dp = new int[size + 1];
        dp[0] = 1;//初始化dp[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];//就是装满容量为size的背包的所有情况
    }
}

474.一和零

力扣题目链接

给你一个二进制字符串数组 strs 和两个整数 m 和 n 。

请你找出并返回 strs 的最大子集的大小，该子集中 最多 有 m 个 0 和 n 个 1 。

如果 x 的所有元素也是 y 的元素，集合 x 是集合 y 的 子集 。

思路

• 确定dp数组（dp table）以及下标的含义

• dp[i] [j]：最多有i个0和j个1的strs的最大子集的大小为dp[i] [j]。

• dp[i] [j] = max(dp[i] [j], dp[i - zeroNum] [j - oneNum] + 1)

• 因为物品价值不会是负数，初始为0，保证递推的时候dpi不会被初始值覆盖。

代码

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        int[][] dp = new int[m + 1][n + 1];
        for (String str : strs) {//遍历字符串数组
            int zeroNum = 0;
            int oneNum = 0;
            for (char ch : str.toCharArray()){//遍历字符串的每个字符
                if (ch == '0') zeroNum++;
                if (ch == '1') oneNum++;//记录每个字符串0和1的数量，每个物品的重量
            }
            for (int i = m; i >= zeroNum; i--){//遍历背包，背包容量是m和n
                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];
    }
}