给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
每个数组中的元素不会超过 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];
}
}