Leetcode-Medium|Leetcode-Medium 416. Partition Equal Subset Sum

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

Input: [1, 5, 11, 5]Output: trueExplanation: The array can be partitioned as [1, 5, 5] and [11].

例子2:
Input: [1, 2, 3, 5]Output: falseExplanation: The array cannot be partitioned into equal sum subsets.

思路 【Leetcode-Medium|Leetcode-Medium 416. Partition Equal Subset Sum】如果两个子集中的元素和相等,那么我们至少可以挖掘两个信息:
  • 如果数组为空,那么应该返回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]

    推荐阅读