leetcode 416 分割等和子集-可以看成背包问题

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

注意:

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

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

输出: true

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

示例 2:

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

输出: false

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

分析:对于该题可以看出是背包问题,即相等于判断能够从数组中选出若干个元素使其和等于sum/2,在使用之前需要对各种明显不符合的情况进行判断,首先长度为1或者0直接false,如果数组元素的和为奇数也返回false,因为数组元素全部为正整数,如果target = sum/2 等于小数,是无法凑出来的。同时我们还需要记录最大元素max,如果最大元素大于target,也是false,因为剩下的无法凑出另一半。

动态规划很重要的一个步骤是确定状态,这里我们创建一个new boolean dp[n][target+1],对于每个dp[i][j]表示从数组下标0-i-1个元素中选出若干个能否凑出其和为target。
状态转移方程为分为两种情况
1.若nums[i] > j,那么肯定不能选择nums[i],如果选了就大于j,无法凑出
此时判断从数组下标0…i-1能否凑出来,dp[i][j] = dp[i-1][j]
2.若nums[i] <= j,此时分为选下标为i的元素和不选下标为i的元素,如果有一种情况能凑出来,dp[i][j]就是true的

代码如下

class Solution {
    public boolean canPartition(int[] nums) {

        int n = nums.length;
        if(n < 2){
            return false;
        }

        int sum = 0;
        int maxVal = nums[0];
        int target = 0; //从数组中选出若干个元素 使其和等于target
        for(int i = 0;i < n;i++){
            sum += nums[i];
            if(nums[i] > maxVal){
                maxVal = nums[i];
            }
        }
        //如果sum = 奇数,那么无法从一个数组中选出若个元素使其等于sum的一半
        //因为数组中全部都是正整数
        if(sum%2 == 1){
            return false;
        }

         target = sum / 2;

         boolean [][] dp = new boolean[n][target + 1];
         //默认初始化全部为0
        //这里表示状态,动态规划很重要的一步就是如何表示状态
        //这里的dp[3][target]表示从下标为0-3个元素中能不能找到若干个元素凑到和的值为target

         if(maxVal > target){

             //一个最大的元素大于总和的一半,说明剩下的值的和不可能凑到target,返回false
             return false;
         }

         //到这里,填充dp数组之前的判断全部做完,依次填充dp数组
         
         //base case 当target为0时,就是对所有的元素都不选,即可,因此为true
         //对于dp[0][nums[0]]也是true的,因为只选第一个元素,能够凑到nums[0],和0,再上一步已经初始化,其他得全部无法得到
        for(int i = 0;i < n;i++){
            dp[i][0] = true;
        }
        //
        dp[0][nums[0]] = true;

        //对于状态转移方程,我们可以考虑两种情况

        // dp[i][j] 情况1 如果nums[i] > j
        //那么 

        //接下载从底向上更新dp
        for(int i = 1; i < n;i++){
            for(int j = 1;j < target + 1;j++){
                if(nums[i] > j){
                    if(dp[i-1][j]){
                        dp[i][j] = true;
                    }
                }else if(nums[i] <= j){
                    if(dp[i-1][j] || dp[i-1][j-nums[i]]){
                        dp[i][j] = true;
                    }
                }
            }
        }

        //到这里 dp[n-1][target] 就是答案
        return dp[n-1][target];

    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值