LeetCode：416. Partition Equal Subset Sum - Python

问题描述：

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

注意:

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

示例 :

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

问题分析：

这个题目给的标签是动态规划，看了一下大神们做的，大多数用的深度优先搜索。在这里就介绍一种简单的方法。思路：求出这些数字中的所有可能的和，放到字典中，然后判断平分值是否在这个字典中，即可。

Python3实现：

# @Time   :2018/08/07
# @Author :LiuYinxing


class Solution:
    def canPartition(self, nums):
        sums = sum(nums)
        if sums % 2 == 1: return False  # 能否被2整除
        all_sum = {0}  # 用于存放所有可能的和
        for n in nums:  # 计算出所有可能的和
            all_sum.update({(v + n) for v in all_sum})
        return (sum(nums) / 2.) in all_sum  # 1/2总和， 是否存在


if __name__ == '__main__':
    nums = [1, 5, 11, 5]
    solu = Solution()
    print(solu.canPartition(nums))

欢迎指正哦。