leetcode 416. 分割等和子集

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

注意:

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

---

示例 1:

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

输出: true

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

示例 2:

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

输出: false

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

---

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

0-1背包问题：

这道题的思路转换比较重要，原问题是分割成两个和相等的子集，可以将其转化为从数组中挑选元素，让元素的和能达到数组总和的一半，那么这就可以看作是一个背包问题。背包中有n个值，每个值都可以选择带或者不带。最后目标是选择m个值让和正好等于目标值。状态dp是一个二维的矩阵，第i行代表第i个值的考虑（可以选择带第i个值或者不带），第j列表示能否让带的所有值之和为j。那么dp[i][j]需要考虑两种情况，如果不带第i个数字，那么dp[i][j]就和dp[i-1][j]一样，而如果带上第i个数字，前i-1个数字的和如果等于j-nums[i]，则带上第i个数字的和为j，所以dp[i][j]=dp[i-1][j-nums[i]]。得到状态转移方程之后我们只需要再考虑初始状态就行，然后需要注意一些边界条件。