动态规划解分割等和子集

问题描述

来源：LeetCode第416题
难度：中等

动态规划解析

这道题要求把数组分为两份，这两份元素的和是否相等。首先我们需要计算所有元素之和sum，然后判断是否为偶数。如果不为偶数，说明不能分割成两个元素之和完成相等的两份，就返回false。如果为偶数，我们只需判断存在一些元素之和为sum/2。如果等于sum/2，剩下的元素之和也为sum/2，说明我们可以把数组分为元素之和相等的两部分，返回true。假设sum/2是一个背包的容量，我们只需要找出一些元素把它放到背包中，如果背包中元素的最大和等于sum/2，说明我们可以把数组分成完成相等的两份。这里可以看成是经典的0-1背包问题。接下来我们在来找一下他的递推公式，定义result[i][j]表示把第i个物品放到容量为j的背包中所获得的的最大值。
第i个物品的值是nums [i -1]：
如果nums [i -1]>j，说明背包容量不够，第i件物品放不进去，所以我们不能选择
第i个物品，那么
result[i][j]=result[i -1][j]；
如 果 nums [i -1]<=j ， 说 明 可 以 把 第 j 个 物 品 放 到 背 包 中 ， 我 们 可 以 选择放也可以
选择不放，取最大值即可，如果放就会占用一部分背包容量，最大价值是
result[i][j]=result[i-1][j-nums[i-1]]+nums[i-1]
如果不放
result[i][j]=result[i -1][j]；
取两者的最大值

递推公式如下

if (j >= nums[i - 1]) {
	 result[i][j] = Math.max(result[i - 1][j], result[i - 1][j - nums[i - 1]] + nums[i - 1]);
 } else {
 	 result[i][j] = result[i - 1][j];
 }

代码

public  boolean canPartition(int[] nums) {
        //数组全部元素的和
        int sum = Arrays.stream(nums).reduce(Integer::sum).getAsInt();
        int target = sum >> 1;
        //如果sum为基数，返回false
        if ((sum &1) == 1) {
            return false;
        }
        int length = nums.length;
        int[][] result = new int[length + 1][target + 1];
        for (int i = 1; i <= length; i++) {
            for (int j = 1; j <= target; j++) {
                if (j >= nums[i - 1]) {
                    result[i][j] = Math.max(result[i - 1][j], result[i - 1][j - nums[i - 1]] + nums[i - 1]);
                } else {
                    result[i][j] = result[i - 1][j];
                }
            }
        }
        return target == result[length][target];
    }

算法复杂度

时间复杂度：O（n^2）
空间复杂度：O（n^2）