070_分割等和子集

知识点:递归、DP备忘录

LeetCode第四百一十六题:https://leetcode-cn.com/problems/partition-equal-subset-sum/submissions/

比较容易想到的朴素解法是使用回溯递归,这里只有选与不选两种操作,所以调用两次递归就行,不需要用for循环;优化方案则是使用dp备忘录,记录每个sum能不能被恰好组合,如果已经记录过则直接返回结果。

语言:GoLang

// DP备忘录,或者记忆化递归
var dp map[int]bool
func canPartition(nums []int) bool {
    sum := 0
    for _, v := range nums {
        sum += v
    }

    if sum % 2 == 1 {
        return false
    }

    dp = map[int]bool{}
    return dfs(nums, 0, sum / 2)
}

func dfs(nums []int, index, sum int) bool {
    if v, ok := dp[sum]; ok {
        return v
    }

    if sum == 0 {
        return true
    }

    if sum > 0 && index < len(nums) {
        // 选或不选
        dp[sum] = dfs(nums, index + 1, sum - nums[index]) || dfs(nums, index + 1, sum)
    }

    return dp[sum]
}
posted @ 2020-07-17 09:01  Cenyol  阅读(83)  评论(0编辑  收藏  举报