题目描述
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:
输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
解法一:动态规划
0-1背包:选取物品的容积总量不能超过规定的总量。思路是:物品一个个先,容量也一点一点放大
- 若nums[i]之前已有元素使得和为j,则dp[i] [j]=true;
- 若选择nums[i]加入,则它们和为j-nums[i];
状态方程:dp[i] [j]=dp[i-1] [j] or dp[i] [j-nums[i]]
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-cF0iSCBF-1594955320348)(D:\leetcode\题目\动态规划\416. 分割等和子集.assets\image-20200717104124807.png)]
code 1
class Solution {
public boolean canPartition(int[] nums) {
int n=nums.length;
int sum=0;
for(int i:nums){
sum+=i;
}
if(sum%2==1||sum==0){
return false;
}
int value=sum/2;
boolean[][] ans = new boolean[n][value+1];
//填充第一行,nums[0]对应的位置
if(nums[0]<=value){
ans[0][nums[0]]=true;
}
for(int i=1;i<n;i++){
for(int j=0;j<=value;j++){
//先填入上一个结果
ans[i][j]=ans[i-1][j];
//刚好得到结果
if(nums[i]==j){
ans[i][j]=true;
continue;
}
//配合之前选择得到结果
if(nums[i]<j){
ans[i][j]=ans[i-1][j]||ans[i-1][j-nums[i]];
}
}
}
return ans[n-1][value];
}
}
code 2 剪枝
class Solution {
public boolean canPartition(int[] nums) {
int n=nums.length;
int sum=0;
for(int i:nums){
sum+=i;
}
if(sum%2==1||sum==0){
return false;
}
int value=sum/2;
boolean[][] ans = new boolean[n][value+1];
//填充第一行,nums[0]对应的位置
if(nums[0]<=value){
ans[0][nums[0]]=true;
}
for(int i=1;i<n;i++){
for(int j=0;j<=value;j++){
//先填入上一个结果
ans[i][j]=ans[i-1][j];
//配合之前选择得到结果
if(nums[i]<=j){
ans[i][j]=ans[i-1][j]||ans[i-1][j-nums[i]];
}
//进行剪枝
if(ans[i][value]==true){
return true;
}
}
}
return ans[n-1][value];
}
}
复杂度同上
code 3 压缩一维
- 当前行总是参考了它上面一行 “头顶上” 那个位置和“左上角”某个位置的值。因此,我们可以只开一个一维数组,从后向前依次填表即可。
class Solution {
public boolean canPartition(int[] nums) {
int n=nums.length;
int sum=0;
for(int i:nums){
sum+=i;
}
if(sum%2==1||sum==0){
return false;
}
int value=sum/2;
boolean[] ans = new boolean[value+1];
//填充第一行,nums[0]对应的位置
if(nums[0]<=value){
ans[nums[0]]=true;
}
for(int i=1;i<n;i++){
//nums[i]<=j为了配合j-nums[i]>=0
for(int j=value;nums[i]<=j;j--){
if(ans[value]){
return true;
}
ans[j]=ans[j]||ans[j-nums[i]];
}
}
return ans[value];
}
}