leetcode416.分割等和子集「大众点评面试题」

1.题目描述

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

注意:

每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:

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

输出: true

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

示例 2:

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

输出: false

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

2.解题思路

参考链接：https://leetcode-cn.com/problems/partition-equal-subset-sum/solution/0-1-bei-bao-wen-ti-xiang-jie-zhen-dui-ben-ti-de-yo/

 

3.代码实现

class Solution(object):
    def canPartition(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        if (sum(nums))%2==1:
            return False
        target=(sum(nums))/2
        n=len(nums)
        dp=[[False for j in range(target+1)] for i in range(n+1)]
        dp[0][0] = True
        nums.insert(0,0)
        # #注意要先对第1个物品初始化
        for j in range(1,target + 1):
             if nums[1] == j:
                 dp[1][j] = True
        for i in range(2,n+1):
            for j in range(1,target+1):
                # 注意判断
                if nums[i]<=j:
                    dp[i][j]=dp[i-1][j-nums[i]] or dp[i-1][j]
                else:
                    dp[i][j]=dp[i-1][j]
        return dp[n][target]
                    
        
        

优化：参考字节跳动面试题.背包系列「01背包、完全背包、二维费用背包、多重背包」

class Solution(object):
    def canPartition(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        if (sum(nums))%2==1:
            return False
        target=(sum(nums))/2
        n=len(nums)
        dp=[False for j in range(target+1)]
        dp[0] = True
        nums.insert(0,0)
        # #注意要先对第1个物品初始化
        for j in range(1,target + 1):
             if nums[1] == j:
                 dp[j] = True
                 break
        for i in range(2,n+1):
            for j in range(target,0,-1):
                # 注意判断
                if nums[i]<=j:
                    dp[j]=dp[j-nums[i]] or dp[j]
        return dp[target]