1. 问题描述:
给你一个二进制字符串数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ones-and-zeroes
2. 思路分析:
分析题目可以知道这是一道无选择顺序的组合性问题,每一个字符串只能够选或者不选,存在两个花费分别是0的数目和1的数目的最大值,所以属于经典的二维费用的零一背包问题,将每一个字符串看成是一个物品,0和1的数目的限制可以看成是体积和重量的限制,我们需要求解在不超过体积和重量的前提下能够获得的最大价值(每一个字符串的价值为1),使用二维费用的零一背包问题的模板解决即可;我们可以定义一个二维数组dp,其中dp[i][j]表示0的数目不超过i,1的数目不超过j的最大子集的长度,使用三层循环枚举,第一层循环枚举第i个字符串,第二层循环枚举0的数目(相当于枚举的是体积),第三层循环枚举1的数目即可(相当于枚举的是重量)。
3. 代码如下:
from typing import List
class Solution:
# 二维费用的零一背包问题(裸题)
def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
dp = [[0] * (n + 1) for i in range(m + 1)]
for s in strs:
zero, one = s.count("0"), s.count("1")
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]