提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
题目描述
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-equal-subset-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题过程
解题思路
动态规划:由背包问题演化而来。根据题意转化一下问题:找到数组中加和值为数组总和的一半的方案。dp[i][j]表示0–i中是否存在求和方案恰好等于j。
处理特殊条件:
1、当数组长度小于2时,数组不可能分割为两部分,直接return false;
2、数组总和sum为奇数,也不可能,直接return false,(假设分割为两部分,那两个数求和必然是偶数的);
3、如果数组里的最大值比target大的话,那么其他值加起来肯定比target小,这样就不能将数组分割成两部分了,直接返回false;
4、dp[i][0]是false,因为正整数求和结果不可能为0的;
5、dp[0][nums[0]]=true,因为前面0个数等到结果为nums[0]是事实。
转移方程:
1、如果nums[i] <= j,那么nums[i]可以被选也可以不被选,
dp[i][j] = dp[i-1][j],不被选
dp[i][j] = dp[i-1][j-nums[i]],被选
如果nums[i] > j,那么nums[i]必然不被选,因为一旦选了dp[i][j]会变矛盾,
dp[i][j] = dp[i-1][j-1]
综合起来就是
dp[i][j] =
dp[i-1][j-1] | dp[i-1][j-nums[i]],nums[i] <= j;
dp[i-1][j-1],nums[i] > j。
class Solution {
public boolean canPartition(int[] nums) {
if (nums.length < 2){
return false;
}
int sum = 0;
int max = 0;
for (int i = 0; i < nums.length; i++) {
sum += nums[i];
max = Math.max(nums[i], max);
}
if (sum %2 != 0){
return false;
}
int target = sum / 2;
//如果数组里的最大值比target大的话,那么其他值加起来肯定比target小,这样就不能将数组分割成两部分了
if (max > target){
return false;
}
boolean [][]dp = new boolean[nums.length][target + 1];
for (int i = 0; i < nums.length; i++) {
dp[i][0] = false;
}
dp[0][nums[0]] = true;
for (int i = 1; i < nums.length; i++) {
for (int j = 0; j < target + 1; j++) {
if (nums[i] <= j){
//此时nums[i]可被选和不被选
//不被选上,就要看前面的0--i-1是否有方案使得加和结果等于j
//被选上,则看前面的0--i-1是否有方案加和结果等于j-nums[i],若有即表明0--i有方案加和结果等于j
dp[i][j] = dp[i - 1][j] | dp[i - 1][j - nums[i]];
}else {
//此时的nums[i]>j,若nums[i]被选了,加和结果肯定超过j,所以nums[i]必然不能选择
//所以只能看前面的0--i-1是否有方案加和结果为j
dp[i][j] = dp[i - 1][j];
}
}
}
return dp[nums.length - 1][target];
}
}