【重点】【背包】【DP】416.分割等和子集

原创已于 2025-08-09 17:26:23 修改 · 455 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#dp

于 2023-12-27 11:01:32 首次发布

力扣Top100 同时被 2 个专栏收录

100 篇文章

订阅专栏

labuladong的算法小抄笔记

85 篇文章

订阅专栏

文章介绍了如何用Java实现基于0-1背包问题的Solution类，通过动态规划方法解决数组元素分组使总和等于目标值的问题，关键在于理解并运用dp数组的构建过程。

题目

法1：dp

基础解法，必须掌握！！！
本质是『0-1背包』问题，背包问题大总结

Python

class Solution:
    def canPartition(self, nums: List[int]) -> bool:
        n = len(nums)
        if sum(nums) % 2 == 1 or max(nums) > sum(nums) / 2:
            return False
        target = sum(nums) // 2
        dp = [[False] * (target + 1) for _ in range(n)] # dp[i][j]表示[0,i]之间的数能否组成j
        for i in range(n):
            dp[i][nums[i]] = True
            dp[i][0] =  True
        
        for i in range(1, n):
            for j in range(1, target+1):
                if nums[i] > j:
                    dp[i][j] = dp[i-1][j] # 不选择i
                else:
                    dp[i][j] = dp[i-1][j] or dp[i-1][j-nums[i]] # 不选 or 选择 
        
        return dp[n-1][target]

Java

class Solution {
    public boolean canPartition(int[] nums) {
        if (nums.length < 2) {
            return false;
        }
        int sum = 0, maxVal = 0;
        for (int i : nums) {
            sum += i;
            maxVal = i > maxVal ? i : maxVal;
        }
        if (sum % 2 == 1 || maxVal > sum / 2) {
            return false;
        }

        int n = nums.length, target = sum / 2;
        boolean[][] dp = new boolean[n][target + 1]; // dp[i][j]表示能否从[0, i]组合sum为j
        for (int i = 0; i < nums.length; ++i) {
            dp[i][nums[i]] = true;
            dp[i][0] = true;
        }

        for (int i = 1; i < nums.length; ++i) {
            for (int j = 1; j <= target; ++j) {
                if (j < nums[i]) {
                    dp[i][j] = dp[i - 1][j];
                } else {
                    dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i]]; // 取nums[i] or not
                }
            }
        }

        return dp[n - 1][target];
    }
}