eBay2021算法真题：划分和相等的子集

原创于 2021-06-11 09:37:08 发布 · 344 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #leetcode #动态规划 #字节跳动

leetcode/lintcode每天一题专栏收录该内容

107 篇文章

订阅专栏

本文探讨了如何解决一个特定的计算机科学问题：给定一个包含正整数的非空数组，判断该数组是否可以被划分为两个子集，使得这两个子集的元素之和相等。文章详细介绍了通过01背包问题的转换来解决这一问题的方法。

描述
给一只含有正整数的非空数组, 找到这个数组是否可以划分为两个元素和相等的子集。
所有数组元素不超过100.
数组大小不超过200.

在线评测地址

样例 1:

输入: nums = [1, 5, 11, 5], 
输出: true
解释:
2个子集: [1, 5, 5], [11]

样例 2:

输入: nums = [1, 2, 3, 9], 
输出: false

解题思路
等价与背包问题，能否背到总价值的一半。 01背包即可。

源代码

class Solution(object):
    def canPartition(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        length = len(nums)
        mySum =  sum(nums);
        if mySum % 2 == 1:
            return False
        mySum /= 2
        dp = [False for x in range(20000)]
        dp[0] = True
        for i in range(0,length):
            for j in range(mySum, nums[i] - 1, -1):
                dp[j] |= dp[j - nums[i]]
        return dp[mySum]

更多题解参考