问题描述
来源: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)