Leetcode每日一题——416.分割等和子集。套用0-1背包。dp四部曲

题目链接:

力扣

题目描述:

给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。

示例 1:

输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:

输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
 

提示:

1 <= nums.length <= 200
1 <= nums[i] <= 100

解析:

题目的意思就是从nums中选取元素,使选取的元素之和=sum/2.

与0-1背包的思路类Leetcode.ipynb:从一堆物品中选择一些放到背包中,使背包中的物品满足一定条件

对比0-1背包,这道题中的sum/2、元素之和、数组里的元素、数组里的元素之和分别相当于

0-1背包中的背包体积、背包容量、价值、背包里物品的价值之和的最大值

dp四部曲:

确定dp数组:dp[j]表示元素之和为j时(相当于连续的和,即这里的和是从0到j连续的),数组中的选取的元素的最大和(相当于离散的和,即这里的和是由数组中的元素组成的),需要注意的是,dp[j]<=j

确定递推公式:dp[j] = max{dp[j], dp[j - nums[i]] + nums[i]},决策第i个元素时的公式:第一项表示不去第i个元素,第二项表示取第i个元素

初始化dp数组: 初始化全部为0就好了,因为这时候还没有开始决策,而且0也是nums中的最小值,可以被其他更大的值覆盖的

确定遍历顺序:遍历nums数组时顺序遍历,遍历j(也就是元素和)时逆序,这个顺序和0-1背包的遍历顺序是一样的

需要注意的是,当dp[sum/2] = sum/2时就说明可找到子集凑成sum/2.。实际上dp[j]=j就说明可以找到子集凑成j了

代码如下:

class Solution:
    def canPartition(self, nums: List[int]) -> bool:
        '''
        题目的意思就是从nums中选取元素,使选取的元素之和=sum/2.
        与0-1背包的思路类Leetcode.ipynb:从一堆物品中选择一些放到背包中,使背包中的物品满足一定条件
        对比0-1背包,这道题中的sum/2、元素之和、数组里的元素、数组里的元素之和分别相当于
        0-1背包中的背包体积、背包容量、价值、背包里物品的价值之和的最大值
        dp四部曲:
        确定dp数组:dp[j]表示元素之和为j时(相当于连续的和,即这里的和是从0到j连续的),数组中的选取的元素的最大和(相当于离散的和,即这里的和是由数组中的元素组成的),需要注意的是,dp[j]<=j
        确定递推公式:dp[j] = max{dp[j], dp[j - nums[i]] + nums[i]},决策第i个元素时的公式:第一项表示不去第i个元素,第二项表示取第i个元素
        初始化dp数组: 初始化全部为0就好了,因为这时候还没有开始决策,而且0也是nums中的最小值,可以被其他更大的值覆盖的
        确定遍历顺序:遍历nums数组时顺序遍历,遍历j时逆序,这个顺序和0-1背包的遍历顺序是一样的
        需要注意的是,当dp[sum/2] = sum/2时就说明可找到子集凑成sum/2.。实际上dp[j]=j就说明可以找到子集凑成j了
        
        '''
        if(sum(nums) % 2 == 1):
            return False
        target = sum(nums) // 2
        dp = [0] * (target + 1)#初始化dp数组
        for i in range(len(nums)):
            for j in range(target, nums[i] - 1, -1):
                dp[j] = max(dp[j], dp[j - nums[i]] + nums[i])
        return dp[target] == target

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
题目描述: 给定一个只包含正整数的非空数组,是否可以将这个数组分成两个子集,使得两个子集的元素和相等。 示例: 输入:[1, 5, 11, 5] 输出:true 解释:数组可以分割成 [1, 5, 5] 和 [11]。 解题思路: 这是一道经典的 0-1 背包问题,可以使用动态规划或者回溯算法解决。 使用回溯算法,需要定义一个 backtrack 函数,该函数有三个参数: - 数组 nums; - 当前处理到的数组下标 index; - 当前已经选择的元素和 leftSum。 回溯过程中,如果 leftSum 等于数组元素和的一半,那么就可以直接返回 true。如果 leftSum 大于数组元素和的一半,那么就可以直接返回 false。如果 index 到达数组末尾,那么就可以直接返回 false。 否则,就对于当前元素,有选择和不选择两种情况。如果选择当前元素,那么 leftSum 就加上当前元素的值,index 就加 1。如果不选择当前元素,那么 leftSum 不变,index 也加 1。最终返回所有可能性的结果中是否有 true。 Java 代码实现: class Solution { public boolean canPartition(int[] nums) { int sum = 0; for (int num : nums) { sum += num; } if (sum % 2 != 0) { return false; } Arrays.sort(nums); return backtrack(nums, nums.length - 1, sum / 2); } private boolean backtrack(int[] nums, int index, int leftSum) { if (leftSum == 0) { return true; } if (leftSum < 0 || index < 0 || leftSum < nums[index]) { return false; } return backtrack(nums, index - 1, leftSum - nums[index]) || backtrack(nums, index - 1, leftSum); } }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

翔空中,策人生

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值