给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:
输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-equal-subset-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
经过分析可得:
就是求是否能从数组中找到 sum/2
public boolean canPartition(int[] nums) {
int sum = 0;
for (int num : nums) {
sum += num;
}
if (sum % 2 != 0) return false;//整数相加不可能得小数
int target = sum / 2;
boolean[][] dp = new boolean[nums.length][target + 1];
dp[0][0] = true;
for (int i = 0; i < nums.length; i++) {
for (int j = 0; j < target + 1; ++j) {
if (i == 0) {
if (nums[i] == j) {
dp[i][j] = true;
}
} else {
dp[i][j] = dp[i - 1][j] || (j >= nums[i] && dp[i - 1][j - nums[i]]);
}
}
}
return dp[nums.length - 1][target];
}
二维改一维:
由于在二维动态规划数组中,当前值依赖上一行的值,以及上一行的左边的位置。所以要从右往左遍历,避免影响。
public boolean canPartition2(int[] nums) {
int sum = 0;
for (int num : nums) {
sum += num;
}
if (sum % 2 != 0) return false;//整数相加不可能得小数
int target = sum / 2;
boolean[] dp = new boolean[target + 1];
dp[0] = true;
for (int i = 0; i < nums.length; i++) {
for (int j = target; j >= nums[i]; j--) {
// 要当前值nums[i] 和 不要当前值nums[i]
dp[j] = dp[j] || dp[j - nums[i]];
}
}
return dp[target];
}
同类型题目: