题目链接:
题目描述:
给你一个二进制字符串数组 strs 和两个整数 m 和 n 。
请你找出并返回 strs 的最大子集的长度,该子集中 最多 有 m 个 0 和 n 个 1 。
如果 x 的所有元素也是 y 的元素,集合 x 是集合 y 的 子集 。
示例 1:
输入:strs = ["10", "0001", "111001", "1", "0"], m = 5, n = 3
输出:4
解释:最多有 5 个 0 和 3 个 1 的最大子集是 {"10","0001","1","0"} ,因此答案是 4 。
其他满足题意但较小的子集包括 {"0001","1"} 和 {"10","1","0"} 。{"111001"} 不满足题意,因为它含 4 个 1 ,大于 n 的值 3 。
示例 2:
输入:strs = ["10", "0", "1"], m = 1, n = 1
输出:2
解释:最大的子集是 {"0", "1"} ,所以答案是 2 。
提示:
1 <= strs.length <= 600
1 <= strs[i].length <= 100
strs[i] 仅由 '0' 和 '1' 组成
1 <= m, n <= 100
解析如下:
求含有m个0和n个1的子集的最大长度。即从数组中选出特定的几个物品,且这些物品满足一定的条件(数量最多)
那么m和n就对应0-1背包的容量了,而且这还是个二维的背包,字符串元素0和 1的数目对应物品重量,物品的价值对应m个0,n个1时的子集长度。
dp四部曲
确定dp数组:dp[m][n]表示最多可以有m个0和n个1时(背包容量)的最大子集长度
确定递推公式:dp[m][n] = max(dp[m][n], dp[m-str[i]的0的个数][n-str[i]的1的个数] + 1)2种情况对应子集不取和取str[i]
初始化dp数组:dp[0][0] = 0表示当容量为0个0和0个1时子集长度为0
确定遍历顺序:同0-1背包问题,外层顺序,内层逆序
代码如下:
class Solution:
def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
dp = [[0] * (n + 1) for i in range(m + 1)]
dp[0][0] = 0#初始化
for i in range(len(strs)):
zero = strs[i].count('0')#m
one = strs[i].count('1')#n
for j in range(m, zero - 1, -1):
for k in range(n, one - 1, -1):
dp[j][k] = max(dp[j][k], dp[j - zero][k - one] + 1)
return dp[m][n]