题目链接:
题目描述:
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 100
解析:
题目的意思就是从nums中选取元素,使选取的元素之和=sum/2.
与0-1背包的思路类Leetcode.ipynb:从一堆物品中选择一些放到背包中,使背包中的物品满足一定条件
对比0-1背包,这道题中的sum/2、元素之和、数组里的元素、数组里的元素之和分别相当于
0-1背包中的背包体积、背包容量、价值、背包里物品的价值之和的最大值
dp四部曲:
确定dp数组:dp[j]表示元素之和为j时(相当于连续的和,即这里的和是从0到j连续的),数组中的选取的元素的最大和(相当于离散的和,即这里的和是由数组中的元素组成的),需要注意的是,dp[j]<=j
确定递推公式:dp[j] = max{dp[j], dp[j - nums[i]] + nums[i]},决策第i个元素时的公式:第一项表示不去第i个元素,第二项表示取第i个元素
初始化dp数组: 初始化全部为0就好了,因为这时候还没有开始决策,而且0也是nums中的最小值,可以被其他更大的值覆盖的
确定遍历顺序:遍历nums数组时顺序遍历,遍历j(也就是元素和)时逆序,这个顺序和0-1背包的遍历顺序是一样的
需要注意的是,当dp[sum/2] = sum/2时就说明可找到子集凑成sum/2.。实际上dp[j]=j就说明可以找到子集凑成j了
代码如下:
class Solution:
def canPartition(self, nums: List[int]) -> bool:
'''
题目的意思就是从nums中选取元素,使选取的元素之和=sum/2.
与0-1背包的思路类Leetcode.ipynb:从一堆物品中选择一些放到背包中,使背包中的物品满足一定条件
对比0-1背包,这道题中的sum/2、元素之和、数组里的元素、数组里的元素之和分别相当于
0-1背包中的背包体积、背包容量、价值、背包里物品的价值之和的最大值
dp四部曲:
确定dp数组:dp[j]表示元素之和为j时(相当于连续的和,即这里的和是从0到j连续的),数组中的选取的元素的最大和(相当于离散的和,即这里的和是由数组中的元素组成的),需要注意的是,dp[j]<=j
确定递推公式:dp[j] = max{dp[j], dp[j - nums[i]] + nums[i]},决策第i个元素时的公式:第一项表示不去第i个元素,第二项表示取第i个元素
初始化dp数组: 初始化全部为0就好了,因为这时候还没有开始决策,而且0也是nums中的最小值,可以被其他更大的值覆盖的
确定遍历顺序:遍历nums数组时顺序遍历,遍历j时逆序,这个顺序和0-1背包的遍历顺序是一样的
需要注意的是,当dp[sum/2] = sum/2时就说明可找到子集凑成sum/2.。实际上dp[j]=j就说明可以找到子集凑成j了
'''
if(sum(nums) % 2 == 1):
return False
target = sum(nums) // 2
dp = [0] * (target + 1)#初始化dp数组
for i in range(len(nums)):
for j in range(target, nums[i] - 1, -1):
dp[j] = max(dp[j], dp[j - nums[i]] + nums[i])
return dp[target] == target