leetcode416分割等和子集——又一个背包问题

题目:

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

注意:

  1. 每个数组中的元素不会超过 100
  2. 数组的大小不会超过 200

示例 1:

输入: [1, 5, 11, 5]

输出: true

解释: 数组可以分割成 [1, 5, 5] 和 [11].

 首先可以看出如果数组和为奇数,是无法分割的(自然数乘以二都是偶数)。

然后其中一个基本想法是BFS,这个数字nums【i】我可以选择“加上”或者“不加”,

也就是双重递归。

思路如下:

    def canPartition_(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        nums_sum=sum(nums)
        if nums_sum%2!=0:
            return False
        else:
            nums_sum//=2
        rec=[False]
        def search(nums,i,s=0):
            if i==len(nums) or s>nums_sum:
                return
            if nums[i]+s==nums_sum:
                rec[0]=True
                return
            else:
                search(nums,i+1,s)
                search(nums,i+1,s+nums[i])
        search(nums,0,0)
        return rec[0]

很可惜,这个方法只能AC不到一半的测试用例(可能C++或者Java会更多吧,毕竟没有递归限制)。

那么参考别人的答案后发现了用背包问题来解决的办法。

实际上我也想到过,但是无法转化成对应代码。

    def canPartition(self,nums):
        nums_sum = sum(nums)
        if nums_sum % 2 != 0:
            return False
        else:
            nums_sum //= 2
        dp=[[0 for x in range(nums_sum+1)] for y in range(len(nums))]
        for i in range(nums[0],len(dp[0])):
            dp[0][i]=nums[0]
        for i in range(1,len(nums)):
            for j in range(nums[i],nums_sum+1):
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-nums[i]]+nums[i])
        return dp[-1][-1]==nums_sum

然而这个代码没有通过,哈哈哈哈。

因为python太慢了,而我参考的java代码(写法几乎一样)是可以通过的。

对于这种可以转化为背包问题的问题,还需要多加练习。

  • 0
    点赞
  • 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
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值