LeetCode 416

最新推荐文章于 2024-10-08 10:36:00 发布

m0_60721823

最新推荐文章于 2024-10-08 10:36:00 发布

阅读量399

点赞数 4

文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/m0_60721823/article/details/138237436

版权

文章讨论了如何通过计算数组元素和并应用0-1背包问题的递推公式，判断是否可以将数组分成两部分，使得两部分的元素和相等。作者给出了两种实现方法，包括使用二维数组和优化为一维数组的版本，以及深度优先搜索（DFS）的解决方案。

摘要由CSDN通过智能技术生成

这题判断把数组分成两份，这两份的元素和是否相等。首先我们需要计算数组中所有元素的和sum，然后判断sum是否是偶数：
如果不是偶数，说明不可能分割成完全相等的两份，直接返回false。

如果是偶数，我们只需要判断是否存在一些元素的和等于sum/2，如果等于sum/2，那么剩下的肯定也等于sum/2，说明我们可以把数组分为元素和相等的两部分。

那么这个时候问题就很明朗了，假设sum/2是一个背包的容量，我们只需要找出一些元素把他放到背包中，如果背包中元素的最大和等于sum/2，说明我们可以把数组分成完成相等的两份。这不就是经典的0-1背包问题吗。之前也讲过371，背包问题系列之-基础背包问题，具体可以看下，这里就不在重复介绍。我们在来找一下他的递推公式，定义dp[i][j]表示把第i个物品放到容量为j的背包中所获得的的最大值。

第i个物品的值是nums[i-1]：
如果nums[i-1]>j，说明背包容量不够，第i件物品放不进去，所以我们不能选择第i个物品，那么
dp[i][j]=dp[i-1][j]；

如果nums[i-1]<=j，说明可以把第j个物品放到背包中，我们可以选择放也可以选择不放，取最大值即可，如果放就会占用一部分背包容量，最大价值是

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

如果不放
dp[i][j]=dp[i-1][j]；
取两者的最大值

最终递推公式如下

if (j >= nums[i - 1]) {
    dp[i][j] = Math.max(dp[i - 1][j], dp[i-1][j - nums[i - 1]] + nums[i - 1]);
} else {
    dp[i][j] = dp[i - 1][j];
}

我们来看下最终代码

public boolean canPartition(int[] nums) {
    //计算数组中所有元素的和
    int sum = 0;
    for (int num : nums)
        sum += num;
    //如果sum是奇数，说明数组不可能分成完全相等的两份
    if ((sum & 1) == 1)
        return false;
    //sum除以2
    int target = sum >> 1;
    int length = nums.length;
    int[][] dp = new int[length + 1][target+1];
    for (int i = 1; i <= length; i++) {
        for (int j = 1; j <= target; j++) {
            //下面是递推公式
            if (j >= nums[i - 1]) {
                dp[i][j] = Math.max(dp[i - 1][j], dp[i-1][j - nums[i - 1]] + nums[i - 1]);
            } else {
                dp[i][j] = dp[i - 1][j];
            }
        }
    }
    //判断背包最大是否能存放和为target的元素
    return dp[length][target] == target;
}

我们还可以这样写，二维数组dp是boolean类型，dp[i][j]表示数组中前i个元素的和是否可以组成和为j，很明显dp[0][0]=true，表示前0个元素（也就是没有元素）可以组成和为0。代码如下

public boolean canPartition(int[] nums) {
    //计算数组中所有元素的和
    int sum = 0;
    for (int num : nums)
        sum += num;
    //如果sum是奇数，说明数组不可能分成完全相等的两份
    if ((sum & 1) == 1)
        return false;
    //sum除以2
    int target = sum >> 1;
    int length = nums.length;
    boolean[][]dp = new boolean[length + 1][target+1];
    dp[0][0] = true;//base case
    for (int i = 1; i <= length; i++) {
        for (int j = 1; j <= target; j++) {
            //递推公式
            if (j >= nums[i - 1]) {
                dp[i][j] = (dp[i - 1][j] || dp[i-1][j - nums[i - 1]]);
            } else {
                dp[i][j] = dp[i - 1][j];
            }
        }
    }
    return dp[length][target];
}

我们看到上面二维数组计算的时候当前值只和上面一行有关，所以我们可以把它改成一维的，注意第二个for循环要倒叙，否则会把前面的值给覆盖掉导致结果错误，仔细看一下
dp[j] = (dp[j] || dp[j - nums[i - 1]]);
就明白了，相当于同一行后面的值依赖前面的，如果不是倒叙，前面的值被修改了，在计算后面的就会导致错误。我们来看下代码。

public boolean canPartition(int[] nums) {
    //计算数组中所有元素的和
    int sum = 0;
    for (int num : nums)
        sum += num;
    //如果sum是奇数，说明数组不可能分成完全相等的两份
    if ((sum & 1) == 1)
        return false;
    //sum除以2
    int target = sum >> 1;
    int length = nums.length;
    boolean[] dp = new boolean[target + 1];
    dp[0] = true;//base case
    for (int i = 1; i <= length; i++) {
        //注意这里j要倒叙
        for (int j = target; j >= 1; j--) {
            //递推公式
            if (j >= nums[i - 1]) {
                dp[j] = (dp[j] || dp[j - nums[i - 1]]);
            }
            //else {//这里省略
            // dp[j] = dp[j];
            //}
        }
    }
    return dp[target];
}

DFS解决

每种元素可以选择也可以不选择，只需要判断他所有的可能组合中，元素和是否有等于sum/2的，我们可以把它看做是一棵二叉树，左子节点表示选择当前元素，右子节点表示不选择当前元素，如下图所示，橙色节点表示选择当前元素，蓝色表示不选择。

在这里插入图片描述
我们来看下代码

public boolean canPartition(int[] nums) {
    //计算数组中所有元素的和
    int sum = 0;
    for (int num : nums)
        sum += num;
    //如果sum是奇数，说明数组不可能分成完全相等的两份
    if ((sum & 1) == 1)
        return false;
    return dfs(nums, sum >> 1, 0);
}


private boolean dfs(int[] nums, int target, int index) {
    //targe等于0，说明存在一些元素的和等于sum/2，直接返回true
    if (target == 0)
        return true;
    //如果数组元素都找完了，或者target小于0，直接返回false
    if (index == nums.length || target < 0)


**一、Python所有方向的学习路线**

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照下面的知识点去找对应的学习资源，保证自己学得较为全面。

![img](https://img-blog.csdnimg.cn/1d40facda2b84990b8e1743f5487d455.png)  
![img](https://img-blog.csdnimg.cn/0fc11d4a31bd431dbf124f67f1749046.png)

**二、Python必备开发工具**

工具都帮大家整理好了，安装就可直接上手！![img](https://img-blog.csdnimg.cn/ff266f529c6a46c4bc28e5f895dec647.gif#pic_center)

**三、最新Python学习笔记**

当我学到一定基础，有自己的理解能力的时候，会去阅读一些前辈整理的书籍或者手写的笔记资料，这些笔记详细记载了他们对一些技术点的理解，这些理解是比较独到，可以学到不一样的思路。

![img](https://img-blog.csdnimg.cn/6d414e9f494742db8bcc3fa312200539.png)

**四、Python视频合集**

观看全面零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

![img](https://img-blog.csdnimg.cn/a806d9b941c645858c61d161aec43789.png)

**五、实战案例**

纸上得来终觉浅，要学会跟着视频一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。![img](https://img-blog.csdnimg.cn/a353983317b14d3c8856824a0d6186c1.png)

**六、面试宝典**

![在这里插入图片描述](https://img-blog.csdnimg.cn/97c454a3e5b4439b8600b50011cc8fe4.png)

![在这里插入图片描述](https://img-blog.csdnimg.cn/111f5462e7df433b981dc2430bb9ad39.png)

###### **简历模板**![在这里插入图片描述](https://img-blog.csdnimg.cn/646863996ac44da8af500c049bb72fbd.png#pic_center)




**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化学习资料的朋友，可以戳这里无偿获取](https://bbs.csdn.net/topics/618317507)**

**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**