Leetcode-Medium 416. Partition Equal Subset Sum

题目描述

给定仅包含正整数的非空数组,查找是否可以将数组划分为两个子集,使得两个子集中的元素总和相等。
例子1:

Input: [1, 5, 11, 5]

Output: true

Explanation: The array can be partitioned as [1, 5, 5] and [11].

例子2:

Input: [1, 2, 3, 5]

Output: false

Explanation: The array cannot be partitioned into equal sum subsets.

思路

如果两个子集中的元素和相等,那么我们至少可以挖掘两个信息:

  • 如果数组为空,那么应该返回False
  • 如果数组元素相加的和为奇数时,应该范围False。
    因为两个相等的数相加必为偶数(full_sum=target+target=2*target=2*target)
    target=sum(nums)//2=sum(nums)>>1(右移一位相当于整除)

思路1 :找出所有可能子集的和,判断 target是否出现在possible_sums
思路2:动态规划:[LeetCode] Partition Equal Subset Sum 相同子集和分割

定义一个一维的dp数组,其中dp[i]表示原数组是否可以取出若干个数字,其和为i。那么我们最后只需要返回dp[target]就行了。初始化dp[0]为true,由于题目中限制了所有数字为正数,那么就不用担心会出现和为0或者负数的情况。关键问题就是要找出状态转移方程了,我们需要遍历原数组中的数字,对于遍历到的每个数字nums[i],需要更新dp数组,我们的最终目标是想知道dp[target]的boolean值,就要想办法用数组中的数字去凑出target,因为都是正数,所以只会越加越大,那么加上nums[i]就有可能会组成区间 [nums[i], target] 中的某个值,那么对于这个区间中的任意一个数字j,如果 dp[j - nums[i]] 为true的话,说明现在已经可以组成 j-nums[i] 这个数字了,再加上nums[i],就可以组成数字j了,那么dp[j]就一定为true。如果之前dp[j]已经为true了,当然还要保持true,所以还要‘或’上自身,于是状态转移方程如下:

dp[j] = dp[j] || dp[j - nums[i]] (nums[i] <= j <= target)

有了状态转移方程,那么我们就可以写出代码了,这里需要特别注意的是,第二个for循环一定要从target遍历到nums[i],而不能反过来,想想为什么呢?因为如果我们从nums[i]遍历到target的话,假如nums[i]=1的话,那么[1, target]中所有的dp值都是true,因为dp[0]是true,dp[1]会或上dp[0],为true,dp[2]会或上dp[1],为true,依此类推,完全使我们的dp数组失效了

代码实现

思路1

class Solution:
    def canPartition(self, nums: List[int]) -> bool:
        if sum(nums) and not sum(nums)%2:
            possible_sums={0}
            for num in nums:
                possible_sums.update({(num+v) for v in possible_sums})
                if sum(nums)>>1 in possible_sums:
                    return True
        return False

思路2

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

你可能感兴趣的:(Leetcode-Medium 416. Partition Equal Subset Sum)