leetcode-416. 分割等和子集

manba_2024-08-19 23:46

题目描述

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

示例 1：

输入：nums = [1,5,11,5]

输出：true

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

示例 2：

输入：nums = [1,2,3,5]

输出：false

解释：数组不能分割成两个元素和相等的子集。

思路

01背包问题，参考：代码随想录

1）先判断数组nums总和是不是偶数，如果是奇数，则直接返回false。因为：数组中的元素都是正整数，如果是奇数的话，除以2，是两个小数的和了。不满足题意

2）题目中找到子集的和是sum(nums)//2，即为存在等和自己的。因为target//2，已经找到一半的子集了，另一半就不用找了

3）初始化：如果题目给的价值都是正整数那么非0下标都初始化为0就可以了，如果题目给的价值有负数，那么非0下标就要初始化为负无穷。

4）dp数组遍历顺序，如果使用一维dp数组，外层循环遍历物品，内层循环遍历背包，且需从大到小遍历背包，已防重复。【不可以先遍历背包再遍历物品】

python 复制代码

class Solution(object):
    def canPartition(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        if sum(nums)%2 != 0:
            return False
        target = sum(nums)//2
        dp = [0] * (target+1)
        for num in nums:
            for j in range(target,num-1,-1):
                dp[j] = max(dp[j],dp[j-num]+num)
        return dp[-1] == target


if __name__ == '__main__':
    s = Solution()
    nums = [1, 5, 11, 5]
    print(s.canPartition(nums))