LeetCode416.分割等和子集

快乐早睡

已于 2024-04-08 08:34:08 修改

阅读量444

点赞数 5

分类专栏： LeetCode 文章标签：算法数据结构 c++ leetcode java

于 2024-04-08 07:43:32 首次发布

本文链接：https://blog.csdn.net/weixin_40909461/article/details/137489545

版权

LeetCode 专栏收录该内容

132 篇文章 0 订阅

订阅专栏

LeetCode416.分割等和子集

题目：

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

示例：

输入： $n u m s = [1, 5, 11, 5]$
输出： $t r u e$
解释： 数组可以分割成 [1, 5, 5] 和 [11] 。

解题思路1：01背包（二维）

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = 0;
        int n = nums.size();
        for(int i = 0; i < n; i++) sum += nums[i];
        if(sum % 2) return false;
        sum /= 2;
        vector<vector<int>> dp(n + 1, vector<int>(sum + 1));
        dp[0][0] = 1;
        for(int i = 1; i <= n; i++)
            for(int j = 0; j <= sum; j++){
                dp[i][j] = dp[i - 1][j];
                if(j >= nums[i - 1]) dp[i][j] = max(dp[i][j], dp[i - 1][j - nums[i - 1]]);
            }
        return dp[n][sum];
    }
};

解题思路2：01背包（一维） C++实现

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = 0, n = nums.size();
        for(auto x: nums){
            sum += x;
        }
        if(sum % 2) return false;
        sum /= 2;
        vector<int> dp(10001, 0);
        for(int i = 1; i < nums.size(); i++)
            for(int j = sum; j >= nums[i]; j--)
                dp[j] = max(dp[j], dp[j - nums[i]] + nums[i]);
        if(dp[sum] == sum) return true;
        return false;
    }
};

解题思路2：01背包（一维） Java实现

class Solution {
    public boolean canPartition(int[] nums) {
        int sum = 0, n = nums.length;
        for(int x : nums){
            sum += x;
        }
        if(sum % 2 == 1) return false;
        sum /= 2;
        int[] dp = new int[sum + 1];
        for(int i = 1; i < n; i++)
            for(int j = sum; j >= nums[i]; j--)
                dp[j] = Math.max(dp[j], dp[j - nums[i]] + nums[i]);
        if(dp[sum] == sum) return true;
        return false;
    }
}