*Leetcode 416. 分割等和子集

Z-Pilgrim

于 2023-02-20 07:10:06 发布

阅读量78

点赞数

分类专栏： LeetCode题解文章标签： leetcode 算法动态规划

本文链接：https://blog.youkuaiyun.com/u011026968/article/details/129117316

版权

LeetCode题解专栏收录该内容

271 篇文章

订阅专栏

该代码实现了一个C++类Solution，包含方法canPartition，通过动态规划方法解决是否能将给定整数数组nums分成两个子集，使得两个子集的和相等。首先计算数组总和，如果总和是奇数则返回false。接着初始化动态规划数组，遍历nums，更新dp状态，当找到和为总和一半的子集时返回true，否则返回false。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

力扣

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

                    if(j == sum/2 && dp[i][j])
                        return true;
                }
            }
        }
        return false;
    }
};