Skip to the content.

leetcode [416] 分割等和子集


Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub


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

注意:

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

示例 1:

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

输出: true

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

示例 2:

输入: [1, 2, 3, 5]

输出: false

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

分割成两个等和子集,可以转换为,选取元素和 = 总和 / 2, 因此dp长度为 总和 / 2,遍历数组,如果dp[目标值 - 当前值] 有解,则有解。

class Solution:
    def canPartition(self, nums: List[int]) -> bool:
        leng = len(nums)
        if len == 0:
            return False
        sum_val = sum(nums)
        if sum_val % 2 != 0:
            return False

        target = sum_val // 2

        dp = [False] * (target + 1)
        dp[0] = True

        if nums[0] <= target:
            dp[nums[0]] = True

        for i in range(1, leng):
            j = target
            while nums[i] <= j:
                if dp[target]:
                    return True

                dp[j] = dp[j] or dp[j - nums[i]]
                j -= 1
        
        return dp[target]

行遍历元素,列遍历目标值

class Solution:
    def canPartition(self, nums: List[int]) -> bool:
        if sum(nums) % 2 or len(nums) == 0: return False
        dp = [[False] * (sum(nums) // 2 + 1) for _ in range(len(nums) + 1)]
        for i in range(len(dp)):
            dp[i][0] = True
        for i in range(1, len(dp)):
            if dp[i][-1]: return True
            for j in range(1, len(dp[0])):
                dp[i][j] = dp[i - 1][j]
                if j >= nums[i - 1]:
                    dp[i][j] = dp[i][j] or dp[i - 1][j - nums[i - 1]]
        return dp[-1][-1]