题目就是要从数组中找一个序列,使他们的和为sum/2。如果暴力硬解,挑选的数组子序列个数不定,复杂度太高,肯定不可取。事实上这是一个01背包问题,对于每个数字,要么选中要么不选中。
具体的,用一个二维数组d[i][j]表示,从前i个元素中挑选子序列是否可以计算出和j。那么我们只要知道当j=sum/2的时候,d[i][j]是否为true。
d[i][j]结果的得出,可以有两种情况
1.d[i-1][j]已经为true,也就是说,已经可以由前i-1个元素中挑选子序列计算出和j,那么d[i][j]自然为true。
2.d[i-1][j-nums[i]]为true,也就是说,前i-1个元素中挑选子序列计算出和j-nums[i],那么加上nums[i]刚好可以完成。
显然,这道题和0-1背包问题很像(https://www.cnblogs.com/Christal-R/p/Dynamic_programming.html)
public class Solution {
public boolean canPartition(int[] nums) {
int sum = 0;
for (int num : nums) sum += num;
if ((sum & 1) == 1) return false;//奇数还是偶数的判断
sum /= 2;
int n = nums.length;
boolean[][] dp = new boolean[n + 1][sum + 1];//dp[i][j]=true表示的能够用前i个数的子集构成j值
dp[0][0] = true;//用0个数构成0是true
for (int i = 1; i <= n; i++) dp[i][0] = true;//用i个数的子集可以构成0
for (int i = 1; i <= sum; i++) dp[0][i] = false;//用0个数不能构成大于0的任何数
for (int i = 1; i <= n ; i++) {
for (int j = 1; j <= sum; j++) {
//dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i - 1]]
//也是2种情况:
//包含nums[i - 1]:dp[i][j] = dp[i - 1][j]
//不包含nums[i - 1]:dp[i][j] = dp[i - 1][j - nums[i - 1]]
dp[i][j] = dp[i - 1][j];
if (j >= nums[i - 1]) dp[i][j] = dp[i][j] || dp[i - 1][j - nums[i - 1]];
}
}
return dp[n][sum];
}
}
class Solution {
public boolean canPartition(int[] nums) {
if (nums == null || nums.length == 0) {
return false;
}
// get sum
int sum = 0;
for (int n : nums) sum += n;
if (sum % 2 == 1) {
return false;
}
return dfs(nums, nums.length - 1, sum / 2);
}
private boolean dfs(int[] nums, int idx, int target) {
if (target == 0) {
return true;
}
if (idx < 0 || target < 0 || target < nums[idx]) {
return false;
}
return dfs(nums, idx - 1, target - nums[idx]) || dfs(nums, idx - 1, target);
}
}