题目
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:
输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
题解
class Solution {
public boolean canPartition(int[] nums) {
int n=nums.length;
int sum=0;
int maxNum=0;
if(n<2){
return false;
}
for(int num:nums){
sum+=num;
maxNum=Math.max(maxNum,num);
}
if(sum%2==1){
return false;
}
if(maxNum>sum/2){
return false;
}
int target=sum/2;
boolean[] dp=new boolean[target+1];
dp[0]=true;
for(int i=0;i<n;i++){
for(int j=target;j>=nums[i];j--){//=??正向的话应该是j从nums[i]开始的,所以反向要加上=
dp[j]=dp[j]|dp[j-nums[i]];//j-nums[i]???
}
}
return dp[target];
}
}
思路:
- 极限条件:若n<2,返回false;若sum为奇数,返回false;若maxnum>sum/2,则返回false。
- 01背包压缩到一维后需要注意的问题:压缩到一维时,要采用逆序。dp[j] = dp[j] || dp[j - nums[i]] 可以理解为 dp[j] (新)= dp[j] (旧) || dp[j - nums[i]] (旧),如果采用正序的话 dp[j - nums[i]]会被之前的操作更新为新值
- boolean数组自动初始化为false,注意边界条件。