【Golang】LeetCode 416. 分割等和子集

416. 分割等和子集

题目描述

思路

这道题其实经过一次问题转换之后将会变得很简单。首先，我们应该对整个数组求和，然后判断和ssum是否为偶数。如果ssum为奇数，它甚至不能够被2整除，就更不存在得到两部分元素和相等的可能性了，直接返回false；否则，target = ssum / 2就是我们要找的元素和，问题转换为：从给定的数组nums中寻找一个子序列，满足子序列的和为target。

由此，我们可以使用动态规划快速解决这个问题，它本质上就是一个数组中每种元素只能用一次的「零钱兑换」问题。我们维护一个dp bool 序列，用来表示当前数值i是否能使用数组当中的元素通过加和的方式表示出来。使用双层循环来对dp进行维护。最外层循环i对nums数组当中的元素进行遍历，因为每个元素只能用一次；第二层循环使用变量j从大到小对target进行遍历，如果满足j >= nums[i]，令dp[j] = dp[j] || dp[j - nums[i]]。

最终返回dp[target]即为最终答案。

Golang 题解

go 复制代码

func canPartition(nums []int) bool {
    // 整个数组的和必须是偶数, 否则直接返回 False
    // 然后要做的就是在数组当中挑选数值, 看一下是否恰好能够满足挑选的数的和 == sum / 2
    ssum := 0
    for _, num := range nums {
        ssum += num
    }

    if ssum % 2 == 1 {
        return false
    }

    target := ssum / 2
    dp := make([]bool, target + 1)
    dp[0] = true

    for i := 0; i < len(nums); i ++ {
        for j := target; j >= nums[i]; j -- {
            dp[j] = dp[j] || dp[j - nums[i]]
        }
    }

    return dp[target]
}