题目:
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
思路:
动态规划
类似于279完全平方数,322零钱兑换,是简单的背包问题,这里是每个元素都只能使用一次,要把数组分成两部分,每部分的和相等,那数组之和一定是偶数,所以如果不是偶数可直接返回假。
目标值是数组之和的一半,然后这个就是包了,数组元素就是可选择的物品,但是每个物品都只能选择一次。
- 初始化长度为目标值加1的动态数组,dp[i]表示当目标值是i的时候,能否找到元素之和为i的子集。初始化dp[0]为True。
- 遍历数组中的数,对于一个数再去遍历动态数组,注意这里是逆序遍历,因为数组中的每个数都只能使用一次,如果是无限次就是从前往后遍历,如果这里也从前往后遍历的话,那判断当前位置所需要的前面的状态就是已经用当前的数更新过的了,不符合要求。
- 再看状态转移公式,如果在当前位置上,也就是当前这个目标和上,要是不用外层循环里遍历的这个数,那它的状态就还是原来的状态,如果用这个数,那当前的状态就要看i-num这个目标和对应的状态了,这个状态是真,这里就是真,这个是假,那用上外层循环这个数也没用。
最后返回目标值所对应的状态。
class Solution(object):
def canPartition(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
n = len(nums)
if n < 2:return False
total = sum(nums)
if total % 2 != 0: return False
target = total // 2
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]