- 分割等和子集
链接🔗: 分割等和子集
💡 题目:给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等 注意: 每个数组中的元素不会超过 100 数组的大小不会超过 200
示例 1:
- 输入: [1, 5, 11, 5]
- 输出: true
- 解释: 数组可以分割成 [1, 5, 5] 和 [11]
示例 2:
- 输入: [1, 2, 3, 5]
- 输出: false
- 解释: 数组不能分割成两个元素和相等的子集.
提示:
- 1 <= nums.length <= 200
- 1 <= nums[i] <= 100
思路
要想把这个题替换成01背包🎒问题需要清楚的4点:
- 背包的总量为 数组Sum / 2
- 背包背的物品重量为数组里面元素的值 价值也是数组里面元素的值
- 背包如果正好装满 则说明找到总和为Sum / 2 的子集
- 物品中不能重复的放入
dp[j] 表示 背包容量为 j 放入物品 时 所背物品的最大重量(价值)
如果背包容量为 target dp[target] 就是 背包容量为 target 时 所能别的最大的重量
如果 dp[target] == target 说明刚好装满背包 以示例一为例 背包容量为 Sum / 2 = 11
递推公式: dp[j] = max(dp[j],dp[j-weight[i]] + value[i])
- 先遍历物品🈚️ 在遍历背包🎒 背包从后往前遍历
dp[j] | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | |
1 | 物品0 | 0 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 |
5 | 物品1 | 0 | 1 | 1 | 1 | 1 | 5 | 6 | 6 | 6 | 6 | 6 | 6 |
11 | 物品2 | 0 | 1 | 1 | 1 | 1 | 5 | 6 | 6 | 6 | 6 | 6 | 11 |
5 | 物品3 | 1 | 1 | 1 | 1 | 1 | 5 | 6 | 6 | 6 | 6 | 10 | 11 |
dp[11] = 11 所以是可以等和分割子集为 True
def canPartition(nums):
target = sum(nums)
if target % 2 != 0 : return False
target_ = target / 2
# 定义 dp数组
dp = [0 for _ in range(target_+1)]
# 初始化为0 不需要操作
# 遍历物品
for i in nums:
# 遍历背包 从后往前遍历 !!!
# for j in range(target_ + 1):
for j in range(target_,i-1,-1):
dp[j] = max(dp[j],dp[j-i] + i)
return dp[target_] == target_
执行结果