2020.10.11
原题:点击此处
考点:01背包变形
题解:
1、如果n<2,必定不能分割为2个子集。
2、计算数组和,如果和不是偶数,必定也不能分割。
3、记录数组最大值,如果数组和除以2后小于最大值,那么也必定不能分割。
dp[i][j] 表示从数组的 [0,i] 下标范围内选取若干个正整数(可以是 0个),是否存在一种选取方案使得被选取的正整数的和等于j。
————————————————————————
dp数组定义:
dp[n][target + 1]
target = sum / 2
初始化dp数组:
1、dp[n][0] = true;
因为背包容量是0时,什么都不拿就已经满了
2、dp[0][nums[0]] = true;
在容量为nums[0]时,必定能装下nums[0]
3、状态转移方程
如果 J < nums[i] ,dp[i][j] = dp[i-1][j];
如果 J >= nums[i] ,
则可以用两个判断条件:
dp[i-1][j] = true;
dp[i-1][ j - nums[i] ] = true;
其中有一个是真 就是ture。
上述条件都不满足,则为false。
时间复杂度(n x target)
空间复杂度(n x target)
代码:
public boolean canPartition(int[] nums){
//前提条件1 n>=2
if(nums.length < 2){
return false;
}
//前提条件2 计算数组和,如果为奇数,则不能分割
int sum = 0;
int max = Integer.MIN_VALUE;
for(int i : nums){
sum += i;
if(i > max){
max = i;
}
}
if(sum % 2 != 0){
return false;
}
//前提条件3 如果总和除以2 小于最大值 说明不能分割
int target = sum/2;
if(target < max){
return false;
}
//下面进行dp
Boolean[][] dp = new Boolean[nums.length][target+1];
//dp初始化 全部设置为false;
for(int i = 0;i< nums.length;i++){
for(int j = 0;j<target+1;j++){
dp[i][j] = false;
}
}
//第0列肯定都是True,因为数字0不需要取
for(int i = 0;i< nums.length;i++){
dp[i][0] = true;
}
//第0行 在num【i】的地方也一定是true
dp[0][nums[0]] = true;
//进行遍历(状态转移方程)
for(int i = 1;i< nums.length;i++){
for(int j = 1;j<target+1;j++){
//如果总和小于当前数字,则取上一行dp
if(j < nums[i]){
dp[i][j] = dp[i-1][j];
}else{
if(dp[i-1][j] == true || dp[i-1][j-nums[i]] == true){
dp[i][j] = true;
}else{
dp[i][j] = false;
}
}
}
}
return dp[nums.length-1][target];
}