416. 分割等和子集
这题换句话说就是是否能找出一个子集,使这个子集的总和等于数组总和的一半,数组里每个元素只能选一次。我们确立一个dp数组,长度为数组总和的half+1,内容为False。索引表示和,索引里的内容表示该数是否能由数组里的元素构成。第一层循环表示选择该数组中的某个数,第二层循环判断加上该数字后能组成的子集和。
class Solution:
def canPartition(self, nums: List[int]) -> bool:
numSum = sum(nums)
if len(nums) < 2:
return False
if numSum % 2 != 0:
return False
half = sum(nums) // 2
maxNum = max(nums)
if maxNum > half:
return False
dp = [False] * (half + 1)
dp[0] = True
for n in nums:
for i in range(half,n-1,-1):
dp[i] |= dp[i-n]
return dp[half]