Leetcode C++《热题 Hot 100-59》416. 分割等和子集

Leetcode C++《热题 Hot 100-59》416. 分割等和子集

1. 题目

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

注意:

每个数组中的元素不会超过 100
数组的大小不会超过 200

示例 1:

输入: [1, 5, 11, 5]

输出: true

解释: 数组可以分割成 [1, 5, 5] 和 [11].

示例 2:

输入: [1, 2, 3, 5]

输出: false

解释: 数组不能分割成两个元素和相等的子集.

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-equal-subset-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 思路

  • 总和等于sum,选择一些数的和使得它等于sum/2
  • dp[i][j] 关键是状态转移方程
  • 不选这个数 dp[i][j] = 1 if dp[i-1][j]=1
  • 选择这个数 dp[i][j] = 1 if dp[i-1][j-nums[i]] = 1
  • 时间复杂度kn,k为数字之和

3. 代码

class Solution {
public:
    bool canPartition(vector<int>& nums) {

        //最多200*100

        bool **dp = new bool*[200];
        int sum = 0;
        for (int i = 0; i < nums.size(); i++) {
            sum += nums[i];
        }
        if (sum%2 == 1)
            return false;

        for (int i = 0; i < 200; i++) 
            dp[i] = new bool[sum+1];
        dp[0][0] = true;
        
        for (int i = 0; i < nums.size(); i++) {
            for (int j = 0; j < sum+1; j++) {
                if (i >0 && dp[i-1][j]) {
                    dp[i][j] = true;
                    continue;
                }
                if (i >0 && j-nums[i] >= 0) {
                    if (dp[i-1][j-nums[i]]) {
                        dp[i][j] = true;
                        continue;
                    }    
                }
                if (i == 0 && j == nums[i])
                    dp[i][j] = true;
                else
                    dp[i][j] = false;
            }
        }

        for (int i = 0; i < nums.size(); i++) {
            if (dp[i][sum/2])
                return true;
        }

        return false;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述: 给定一个包含正整数非空数组,是否可以将这个数组分成两个子集,使得两个子集的元素和相等。 示例: 输入:[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、付费专栏及课程。

余额充值