剑指 Offer II 101. 分割等和子集
给定一个非空的正整数数组 nums ,请判断能否将这些数字分成元素和相等的两部分。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:nums 可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:nums 不可以分为和相等的两部分
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 100
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/NUPfPr
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
方法一:空间复杂度O(nt)
n为数组长度,t为数组总和的一半。
1、找状态转移方程。如果数组可以分割为两个相等的子集,那么数组的和sum一定是偶数,那么我们只需要找到和是数组和的一半的子集即可。设f(i, j)表示数组索引0到 i 的中某几个数字的和等于 j (f(i, j) = true),则f(nums.length, sum / 2)即为所求,本题可以看作背包问题,背包的总容量是sum / 2,可以装在背包里的是数组索引0到 i 中的数,现在找f(i, j)与前面的关系,对于判断能否从前 i 个物品中选择若干物品放满容量为 j 的背包时,对标号 i - 1 的物品有两个选择,一是选择将标号为i - 1的物品放入背包中,如果能从前i - 1个物品中选择若干物品放满容量为j - nums[i - 1]的背包(即f(i - 1, j - nums[i - 1])为true),那么f(i, j)就为true;另外一个选择是不将标号为i - 1的物品放入背包中,如果从前i - 1个物品中选择若干物品放满容量为 j 的背包(即f(i - 1, j) = true),那么f(i, j)也为true。下面考虑i 或 j 为0的情况,当 j 为0时,不需要选择任何物品,f(i, 0)都为true,当i 为0时,没有物品可以装背包,所以当j 大于0,f(0, j)都为false。
2、根据状态转移方程获得dp数组。先填满第一行和第一列,后面的直接遍历填充即可,最后返回dp[nums.length][sum / 2]。
方法二:空间复杂度O(t)
跟前面的题目类似,本题也只用到了dp数组的两层,那么可以优化到一维。内循环选择从右向左遍历的原因是:如果从左向右遍历f(i - 1, j)被f(i, j)替换了之后还可能会使用到,但是如果从右向左遍历f(i - 1, j)不会再次被使用
题解(Java)
方法一:空间复杂度O(nt)
class Solution {
public boolean canPartition(int[] nums) {
int sum = 0;
for (int i = 0; i <