题目描述:给你一个二进制字符串数组 strs 和两个整数 m 和 n 。
请你找出并返回 strs 的最大子集的大小,该子集中 最多 有 m 个 0 和 n 个 1 。
如果 x 的所有元素也是 y 的元素,集合 x 是集合 y 的 子集 。
解题思路:这个也是一个背包问题,只是现在🎒有0
和1
两种容量需要考虑,因此有三维动态规划来考虑,代码如下:
class Solution:
def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
size = len(strs)
nums_0 = []
nums_1 = []
for s in strs:
num_0, num_1 = 0, 0
for c in s:
if c == '0':
num_0 += 1
else:
num_1 += 1
nums_0.append(num_0)
nums_1.append(num_1)
dp = [[0] * (m+1) for _ in range(n+1)]
for k in range(size):
num0, num1 = nums_0[k], nums_1[k]
for i in range(n, num1-1, -1):
for j in range(m, num0-1, -1):
dp[i][j] = max(dp[i][j], dp[i-num1][j-num0]+1)
return dp[n][m]