问题描述:
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
来源:力扣(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];
}