概述
Leetcode416. 分割等和子集
题目:
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
题解:
动态规划:
- 数组长度 n < 2 n<2 n<2,返回false;
- 数组所有元素和 sum textit{sum} sum以及最大元素 maxNum textit{maxNum} maxNum ,如果 sum textit{sum} sum是奇数,返回false;
- 如果 sum textit{sum} sum是偶数, target = sum 2 textit{target}=frac{textit{sum}}{2} target=2sum,需要判断是否可以从数组中选出一些数字,使得这些数字的和等于 target textit{target} target,如果 maxNum > target textit{maxNum}>textit{target} maxNum>target,则除了 maxNum textit{maxNum} maxNum 以外的所有元素之和一定小于 target textit{target} target,因此不可能将数组分割成元素和相等的两个子集,直接返回false;
创建二维数组 dp textit{dp} dp,包含 n n n 行 target + 1 textit{target}+1 target+1列,其中 dp [ i ] [ j ] textit{dp}[i][j] dp[i][j] 表示从数组的 [ 0 , i ] [0,i] [0,i] 下标范围内选取若干个正整数(可以是 0 个),是否存在一种选取方案使得被选取的正整数的和等于 j j j。初始时, dp textit{dp} dp 中的全部元素都是 false text{false} false。
在定义状态之后,需要考虑边界情况。以下两种情况都属于边界情况。
- 如果不选取任何正整数,则被选取的正整数等于 0。因此对于所有 0 ≤ i 0 le i 0≤i < n,都有 dp [ i ] [ 0 ] = true textit{dp}[i][0]=text{true} dp[i][0]=true。
- 当 i = 0 i=0 i=0 时,只有一个正整数 nums [ 0 ] textit{nums}[0] nums[0] 可以被选取,因此 dp [ 0 ] [ nums [ 0 ] ] = true textit{dp}[0][textit{nums}[0]]=text{true} dp[0][nums[0]]=true。
对于 i > 0 i>0 i>0 且 j > 0 j>0 j>0 的情况,如何确定 dp [ i ] [ j ] textit{dp}[i][j] dp[i][j] 的值?需要分别考虑以下两种情况。
如果 j ≥ nums [ i ] j ge textit{nums}[i] j≥nums[i],则对于当前的数字 n u m s [ i ] nums[i] nums[i],可以选取也可以不选取,两种情况只要有一个为 t r u e true true,就有 dp [ i ] [ j ] = true textit{dp}[i][j]=text{true} dp[i][j]=true。
-
如果不选取 nums [ i ] textit{nums}[i] nums[i],则 dp [ i ] [ j ] = dp [ i − 1 ] [ j ] textit{dp}[i][j]=textit{dp}[i-1][j] dp[i][j]=dp[i−1][j];
-
如果选取 n u m s [ i ] nums[i] nums[i],则 dp [ i ] [ j ] = dp [ i − 1 ] [ j − nums [ i ] ] textit{dp}[i][j]=textit{dp}[i-1][j-textit{nums}[i]] dp[i][j]=dp[i−1][j−nums[i]]。
如果
j
<
nums
[
i
]
j < textit{nums}[i]
j<nums[i],则在选取的数字的和等于
j
j
j 的情况下无法选取当前的数字
n
u
m
s
[
i
]
nums[i]
nums[i],因此有
dp
[
i
]
[
j
]
=
dp
[
i
−
1
]
[
j
]
textit{dp}[i][j]=textit{dp}[i-1][j]
dp[i][j]=dp[i−1][j]。
状态转移方程如下:
dp [ i ] [ j ] = { dp [ i − 1 ] [ j ] ∣ dp [ i − 1 ] [ j − nums [ i ] ] , j ≥ nums [ i ] dp [ i − 1 ] [ j ] , j < nums [ i ] textit{dp}[i][j]=begin{cases} textit{dp}[i-1][j]~|~textit{dp}[i-1][j-textit{nums}[i]], & j ge textit{nums}[i] \ textit{dp}[i-1][j], & j < textit{nums}[i] end{cases} dp[i][j]={dp[i−1][j] ∣ dp[i−1][j−nums[i]],dp[i−1][j],j≥nums[i]j<nums[i]
最终得到 dp [ n − 1 ] [ target ] textit{dp}[n-1][textit{target}] dp[n−1][target]即为答案。
java代码:
public boolean canPartition(int[] nums) {
int n = nums.length;
if (n < 2) return false;
int sum = 0;
int maxValue = 0;
for (int num : nums) {
sum += num;
maxValue = Math.max(maxValue, num);
}
//总和是奇数
if (sum % 2 != 0) {
return false;
}
//最大值超过总和的一半
int target = sum / 2;
if (maxValue > target) {
return false;
}
// dp[i][j] : 从[0,i]中选取nums中的若干个正整数使得和为j
boolean[][] dp = new boolean[n][target + 1];
for (int i = 0; i < n; i++) {
//从[0,i]中不选任何正整数
dp[i][0] = true;
}
//i=0时,只有一个正数nums[0]可以被选取
dp[0][nums[0]] = true;
for (int i = 1; i < n; i++) {
int num = nums[i];
for (int j = 1; j <= target; j++) {
if (j >= num) {
dp[i][j] = dp[i - 1][j] || dp[i - 1][j - num];
} else {
dp[i][j] = dp[i - 1][j];
}
}
}
return dp[n - 1][target];
}
最后
以上就是悦耳小松鼠为你收集整理的Leetcode416. 分割等和子集Leetcode416. 分割等和子集的全部内容,希望文章能够帮你解决Leetcode416. 分割等和子集Leetcode416. 分割等和子集所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复