Leetcode416. 分割等和子集-代码随想录

本文介绍了两种不同版本的动态规划方法解决01背包问题,canPartition函数用于判断给定整数数组能否分成两部分,使得它们的和相等。作者在首刷和二刷后的代码中展示了如何利用动态规划求解并优化背包问题。
摘要由CSDN通过智能技术生成

目录

题目:

代码(首刷看解析 2024年2月23日:

代码(二刷看解析 2024年3月10日)

代码(三刷自解 2024年6月26日 go)


题目:


代码(首刷看解析 2024年2月23日:

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        /*因为数值 < 100, length < 200 , 数值 < 20000*/
        vector<int> dp(10001, 0);
        int sum = accumulate(nums.begin(), nums.end(), 0);

        if (sum % 2 == 1) return false;
        int target = sum / 2;

        for (int i = 0; i < nums.size(); ++i) {
            for (int j = target; j >= nums[i]; --j) {
                dp[j] = max(dp[j], (dp[j - nums[i]]) + nums[i]);
            }
        }
        if (dp[target] == target) return true;
        return false;
    }
};

代码(二刷看解析 2024年3月10日)

        01背包不知其所以然

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = accumulate(nums.begin(), nums.end(), 0);
        if (sum % 2 == 1) return false;
        int target = sum / 2;

        vector<int> dp(target + 1, 0);
        for (int i = 0; i < nums.size(); ++i) {
            for (int j = target; j >= nums[i]; --j) {
                dp[j] = max(dp[j - nums[i]] + nums[i], dp[j]);
            }
        }
        return dp[target] == target;
    }
};

代码(三刷自解 2024年6月26日 go)

debug了几次debug对了,思路是没问题,就是小错误需要避免。go的小问题挺多,不知道cpp会不会好一点

func canPartition(nums []int) bool {
    // 两个数组分割和相等,如果和是奇数不可能
    var sum int
    for _,v := range nums {
        sum += v
    }
    if sum % 2 == 1 {
        return false
    }
    // 如果是偶数,则若存在一种分割方式使得 newSum == sum / 2,则true,否则false
    // 01背包: 背包总量w = sum / 2, 物品数量 = len(nums), 单个物品价值 == 单个物品重量 == nums[i]
    // dp[i][j] 表示0~i个物品,总量为w存不存在一种刚刚好的方式 则 dp[i][j] = dp[i - 1][j] or dp[i - 1][j - num[i]] + nums[i]
    dp := make([][]int, len(nums) + 1)
    for k := range dp {
        dp[k] = make([]int, sum / 2 + 1)
    }
    for i := 1; i < len(nums); i++ {
        for j := 0; j <= sum / 2; j++ {
            if j < nums[i] {
                dp[i][j] = dp[i - 1][j]
            } else {
                dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - nums[i]] + nums[i])
            }
        }
    }
    return dp[len(nums) - 1][sum/2] == sum/2
}

题目描述: 给定一个只包含正整数的非空数组,是否可以将这个数组分成两个子集,使得两个子集的元素和相等。 示例: 输入:[1, 5, 11, 5] 输出:true 解释:数组可以分割成 [1, 5, 5] 和 [11]。 解题思路: 这是一道经典的 0-1 背包问题,可以使用动态规划或者回溯算法解决。 使用回溯算法,需要定义一个 backtrack 函数,该函数有三个参数: - 数组 nums; - 当前处理到的数组下标 index; - 当前已经选择的元素和 leftSum。 回溯过程中,如果 leftSum 等于数组元素和的一半,那么就可以直接返回 true。如果 leftSum 大于数组元素和的一半,那么就可以直接返回 false。如果 index 到达数组末尾,那么就可以直接返回 false。 否则,就对于当前元素,有选择和不选择两种情况。如果选择当前元素,那么 leftSum 就加上当前元素的值,index 就加 1。如果不选择当前元素,那么 leftSum 不变,index 也加 1。最终返回所有可能性的结果中是否有 true。 Java 代码实现: class Solution { public boolean canPartition(int[] nums) { int sum = 0; for (int num : nums) { sum += num; } if (sum % 2 != 0) { return false; } Arrays.sort(nums); return backtrack(nums, nums.length - 1, sum / 2); } private boolean backtrack(int[] nums, int index, int leftSum) { if (leftSum == 0) { return true; } if (leftSum < 0 || index < 0 || leftSum < nums[index]) { return false; } return backtrack(nums, index - 1, leftSum - nums[index]) || backtrack(nums, index - 1, leftSum); } }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值