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

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

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

有一堆石头，用整数数组 stones 表示。其中 stones[i] 表示第 i 块石头的重量。

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

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

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

示例 1：

输入：stones = [2,7,4,1,8,1]
输出：1
解释：
组合 2 和 4，得到 2，所以数组转化为 [2,7,1,8,1]，
组合 7 和 8，得到 1，所以数组转化为 [2,1,1,1]，
组合 2 和 1，得到 1，所以数组转化为 [1,1,1]，
组合 1 和 1，得到 0，所以数组转化为 [1]，这就是最优值。

题解：把石头总和分成近似相等的两堆，然后让他们相撞。每个石头只能取一次，就是一个一维的01背包问题。

• dp[j]：装满重量为 j 的背包的最大价值 是dp[j]
• 递推公式： dp[j]=Math.max(dp[j],dp[j-weight[j]]+value[i])
• 初始化： dp[0]=0
• 遍历顺序：一维数组的背包问题，物品从前往后，背包从后往前（保证每个物品只被取一次）

​ for(int i=0;i<n;i++){ //物品

​ for(int j=m;j>0;j–){ //背包

​ …(递推公式)

}

}

• 打印dp数组：（强制for）

代码：

class Solution {
    public int lastStoneWeightII(int[] stones) {
        int sum=0;
        for(int i:stones){
            sum+=i;
        }
        int target=sum/2;
        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-dp[target]-dp[target];
    }
}

494.目标和

给你一个非负整数数组 nums 和一个整数 target 。

向数组中的每个整数前添加 '+' 或 '-' ，然后串联起所有整数，可以构造一个 表达式 ：

• 例如，nums = [2, 1] ，可以在 2 之前添加 '+' ，在 1 之前添加 '-' ，然后串联起来得到表达式 "+2-1" 。

返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。

示例 1：

输入：nums = [1,1,1,1,1], target = 3
输出：5
解释：一共有 5 种方法让最终目标和为 3 。
-1 + 1 + 1 + 1 + 1 = 3
+1 - 1 + 1 + 1 + 1 = 3
+1 + 1 - 1 + 1 + 1 = 3
+1 + 1 + 1 - 1 + 1 = 3
+1 + 1 + 1 + 1 - 1 = 3

题解：也是一个分集合问题，将元素分为加法和减法两个集合。有多少个加法集合或者是多少个减法集合就有多少种方法。

• dp[j] : 装满背包为 j 有 dp[j] 种方法
• 递推公式： dp[j]+=dp[j-nums[i]]
• 初始化： dp[0]=0
• 遍历顺序 ：一维数组遍历方法
• 打印dp数组

代码：

class Solution {
    public int findTargetSumWays(int[] nums, int target) {
        int sum=0;
        for(int i:nums){
            sum+=i;
        }
        //目标大于总和
        if(Math.abs(target)>sum) return 0;
        //不能平分为加法集合
        if((target+sum)%2==1) return 0;

        int bagSize=(target+sum)/2;
        int []dp=new int[bagSize+1];
        dp[0]=1;
        for(int i=0;i<nums.length;i++){  //物品
            for(int j=bagSize;j>=nums[i];j--){   //背包
                dp[j]+=dp[j-nums[i]];
            }
        }
        return dp[bagSize];
    }
}

474.一和零

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

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

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

示例 1：

输入：strs = ["10", "0001", "111001", "1", "0"], m = 5, n = 3
输出：4
解释：最多有 5 个 0 和 3 个 1 的最大子集是 {"10","0001","1","0"} ，因此答案是 4 。
其他满足题意但较小的子集包括 {"0001","1"} 和 {"10","1","0"} 。{"111001"} 不满足题意，因为它含 4 个 1 ，大于 n 的值 3 。

题解：01背包问题。背包的容量有两个维度，也就是在往背包中装物品的时候需要考虑两个方面。

• dp[i]j :满足 i 个0 ， j 个1 时最多能装 dp[i]j个物品
• 递推公式: dp[i]j= Math.max(dp[i-x]j-y +1,dp[i]j);

代码：

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        int [][] dp=new int[m+1][n+1];
        dp[0][0]=0;
        for(String str:strs){   //物品
            int x=0,y=0;
            for(char c:str.toCharArray()){
                if(c=='0'){
                    x++;
                }else {
                    y++;
                }
            }
            for(int i=m;i>=x;i--){  //背包，两个维度，0和1的情况都要满足
                for(int j=n;j>=y;j--){
                    dp[i][j]=Math.max(dp[i][j],dp[i-x][j-y]+1);
                }
            }
        }
        return dp[m][n];
    }
}