分割等和子集 (递归 二维dp 一维dp)

问题描述:

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

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

大体思路:

由于想将数组分割为两个和相同的子集,因此可以先获得数组整体的和,然后该问题可以转化为找数组中是否存在某几个元素之和等于总和的一半。注:若总和为奇数则可以直接返回false,由于数组全为正整数,不会出现0.5的。

递归解法:

若给定一个函数dfs(i,  target)表示从i位置开始能否组成target的值,对于每个i有选和不选两种可能,想要返回true(构成target),要么dfs(i + 1, target)返回true,要么dfs(i + 1, target - nums[i])为true。实现代码入下:

    public boolean dfs(int index, int[] nums, int target){
        if(index == nums.length){
            return target == 0;
        }
        return dfs(index + 1, nums, target) || dfs(index + 1, nums, target - nums[index]);
    }

二维dp:

由于上述递归解法中存在大量的重复计算,而返回值只与index和target的取值有关,因此很容易想到改为二维dp.该解法中dp[i][j] 表示从i位置开始能否构成j。

初值:dp[length][0] = true dp[length][j] = false j != 0

转移函数:dp[i][j] = dp[i + 1][j] || dp[i + 1][j - nums[i]] 

实现代码如下:

    public boolean canPartition(int[] nums) {
        int sum = 0;
        for(int num : nums){
            sum += num;
        }
        if(sum % 2 != 0){
            return false;
        }
        // 二维dp dp[i][j] 表示从i位置开始可以构成j
        boolean[][] dp = new boolean[nums.length + 1][sum / 2 + 1];
        dp[nums.length][0] = true;
        for(int i = nums.length - 1; i >= 0; i--){
            for(int j = 0; j < dp[0].length; j++){
                if(dp[i + 1][j] || (j >= nums[i] && dp[i + 1][j - nums[i]])){
                   dp[i][j] = true; 
                }
            }
        }
        return dp[0][sum / 2];
    }

一维dp:

我们在二维dp中发现dp[i][ : ] 只与dp[i + 1][ : ]有关,因此很容易想到让他们共用一个一维数组,即dp[j] 为从当前位置开始能否构成j。此外我们发现dp[i][j] 的取值只有其正下方 和左下方的值有关,因此应该从右往左依次填表。(若从左至右填,会出现把第i1代当第i+1代)。

初值:dp[0] = true dp[j] = false j != 0

转移函数:dp[j] = dp[j] || dp[j - nums[i]] 

实现代码如下:

    public boolean canPartition(int[] nums) {
        int sum = 0;
        for(int num : nums){
            sum += num;
        }
        if(sum % 2 != 0){
            return false;
        }
        
        // 空间压缩 一维dp[i] 
        boolean[] dp = new boolean[sum / 2 + 1];
        dp[0] = true;
        for(int i = nums.length - 1; i >= 0; i--){
            for(int j = dp.length - 1; j >= 0; j--){
                dp[j] = dp[j] || (j >= nums[i] && dp[j - nums[i]]);
            }
        }
        return dp[sum / 2];
    }

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值