一、题目描述
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
二、解题
动态规划
- 首先这题需要判断所有数的和是否为偶数。
- 使用动态规划, dp[i][j]表示从数组的 [0, i] 这个子区间内挑选一些正整数,每个数只能用一次,使得这些数的和恰好等于 j。
class Solution {
public boolean canPartition(int[] nums) {
int length = nums.length;
if(length<2){
return false;
}
int sum = 0;
for(int i = 0;i<length;i++){
sum += nums[i];
}
if(sum%2==1){
return false;
}
sum = sum/2;
//动态规划
boolean[][] dp = new boolean[length][sum+1];
//第一列全部为true 第一列数据为0 此时是成立的
for(int i = 0;i<length;i++){
dp[i][0] = true;
}
//将第一行的第一个数设置为true
if(nums[0]<sum){
dp[0][nums[0]] = true;
}
//动态规划循环
//从第二行开始循环
for(int i = 1;i<length;i++){
int num = nums[i];
for(int j = 0;j<=sum;j++){
//若当前值大于j值 dp[i][j]=dp[i−1][j];dp值选取上一行值
if(j<num){
dp[i][j] = dp[i-1][j];
}else{
//若当前值小于等于J值 若等于直接等于上一个行的值||上一行的J值-当前值的dp值
dp[i][j] = dp[i-1][j] || dp[i-1][j-num];
}
}
}
return dp[length-1][sum];
}
代码2:
class Solution {
public boolean canPartition(int[] nums) {
int length = nums.length;
if(length<2){
return false;
}
int sum = 0;
for(int i = 0;i<length;i++){
sum += nums[i];
}
if(sum%2==1){
return false;
}
sum = sum/2;
//动态规划
boolean[] dp = new boolean[sum+1];
dp[0] = true;
for(int i = 0;i<length;i++){
for(int j = sum;j>=nums[i];j--){
dp[j] = dp[j] || dp[j-nums[i]];
}
}
return dp[sum];
}
}