【leetcode】416. Partition Equal Subset Sum

题目如下:

解题思路:对于这种判断是否的题目,首先看看动态规划能不能解决。本题可以看成是从nums中任选i个元素,判断其和是否为sum(nums)/2,很显然从nums中任选i个元素的和的取值范围是[0,sum(nums)],这里就可以用一个dp数组来保存nums中任选i个元素的和的取值的和,记dp[v] = 1表示可以从nums中任选i个元素使得其和等于v,dp[v] = 0 则表示不可以。那么nums中任意元素nums[i]来说,只要找出dp[0,sum(nums[0:i-1])] 中所有为1的元素,记为j,可以有dp[nums[i] + j] = 1 if dp[j] = 1。

代码如下:

class Solution(object):
    def canPartition(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        if sum(nums) % 2 == 1:
            return False
        nums.sort()
        dp = [0] * (sum(nums)+1)
        dp[nums[0]] = 1
        total = nums[0]
        for i in range(1,len(nums)):
            for j in range(total+1):
                if dp[j] == 1:
                    dp[j + nums[i]] = 1
            total += nums[i]
        #print dp
        return dp[sum(nums)/2] == 1
原文地址:https://www.cnblogs.com/seyjs/p/9578067.html