416. Partition Equal Subset Sum

最新推荐文章于 2022-10-31 19:20:48 发布

原创最新推荐文章于 2022-10-31 19:20:48 发布 · 187 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

12 篇文章

订阅专栏

Python

3 篇文章

订阅专栏

探讨如何判断一个正整数数组是否可以分为两个子集，使两子集元素之和相等。介绍两种方法：深度优先搜索（DFS）及动态规划（DP）。最终采用DP实现，通过构建二维布尔数组解决此问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a non-empty array containing only positive integers, find if the array can be partitioned into two subsets such that the sum of elements in both subsets is equal.

Note:

Each of the array element will not exceed 100.
The array size will not exceed 200.

Example 1:

Input: [1, 5, 11, 5]

Output: true

Explanation: The array can be partitioned as [1, 5, 5] and [11].

Example 2:

Input: [1, 2, 3, 5]

Output: false

Explanation: The array cannot be partitioned into equal sum subsets.

这道题可以说非常好，我想到两种方法，第一种是dfs，current sum==target sum就return True,否则for i in range(len(nums)),每次选一个num[i]加到current sum里，然后nums变为nums[:i]+nums[i+1:]，进行递归。

可惜啊，这方法超时了。

于是乎，咱们直接上dp。说到dp这题是subset sum的变种题。

图表：

横坐标是amount也就是target sum，纵坐标是nums,dp规则见代码。就这样~

class Solution(object):
    def canPartition(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        totalsum=sum(nums)
        if totalsum%2:
            return False
        targetsum=totalsum/2
        
        mem=[[False for _ in range(targetsum+1)] for _ in range(len(nums))]
        for r in range(len(nums)):
            mem[r][0]=True
        
        for r in range(len(nums)):
            for c in range(1,targetsum+1):
                if nums[r]>c:
                    mem[r][c]=mem[r-1][c]
                else:
                    if mem[r-1][c]:
                        mem[r][c]=True
                    else:
                        mem[r][c]=mem[r-1][c-nums[r]]
        
        return mem[-1][-1]