416. 分割等和子集
1.题目
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意: 每个数组中的元素不会超过 100 数组的大小不会超过 200
2.思路
化背包问题:
1)dp数组要素:背包限制,往背包里填充物品,物品的数量和价值
2)背包类型:01背包物品要么选要么不选,完全背包要么选要么可选很多次
3)情景代入:
(1)背包的体积为sum / 2
(2)背包要放入的商品(集合里的元素)重量为元素的数值,价值也为元素的数值
(3)背包如果正好装满,说明找到了总和为 sum / 2 的子集。
(4)背包中每一个元素是不可重复放入。
重点:明确01背包使用场景,且可以解决该问题,其次各要素表示含义
class Solution:
def canPartition(self, nums: List[int]) -> bool:
# sum/2为背包最大容量,当添加的数之和=sum/2时返回true
total = sum(nums)
if total % 2 != 0:
return False
bag_weight = int(total / 2)
dp = [0] * (bag_weight + 1) # dp[i]表示最大和不超过i的数的最大和
for u in nums:
for j in range(bag_weight, -1, -1):
if j < u: # 容量小于数的大小
break
dp[j] = max(dp[j], dp[j - u] + u)
if dp[bag_weight] == bag_weight: # 当该式成立时,则说明存在该组合
return True
return False
474. 一和零
1.题目
给你一个二进制字符串数组 strs 和两个整数 m 和 n 。请你找出并返回 strs 的最大子集的大小,该子集中 最多 有 m 个 0 和 n 个 1 。如果 x 的所有元素也是 y 的元素,集合 x 是集合 y 的 子集 。
2.思路:
困难点:想出来了dp[i][j]的定义,但不知如何推导递推公式,没有办法套到01背包里
突破点:
(1)m 和 n相当于是一个背包,两个维度的背包
(2)字符串的zeroNum和oneNum相当于物品的重量(weight[i]),字符串本身的个数相当于物品的价值(value[i])
class Solution:
def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
# dp[i][j]表示0的数目不超过i、1的数目不超过j时的最大子集数目
k = len(strs)
dp = [[0] * (n+1) for _ in range(m+1)]
for t in range(k): # 遍历每个物品
zeronum, onenum = 0, 0
for u in strs[t]:
if u == '0':
zeronum += 1
else:
onenum += 1
for i in range(m, zeronum - 1, -1): # 遍历两个维度的背包容量,且从大到小遍历
for j in range(n, onenum - 1, -1):
dp[i][j] = max(dp[i - zeronum][j - onenum] + 1, dp[i][j])
return dp[m][n]
还剩了两种类型的背包应用问题:
1049. 最后一块石头的重量 II
494. 目标和
有点难度,可以有时间再补!