给你一个二进制字符串数组 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
这题就是典型的01背包问题,我们可以用动态规划五部曲来解决,首先我们要定义一个数组dp[j][k],最多有i个0和j个1的strs的最大子集的大小为dp[i][j]。接下来我们要初始化数组dp[0][0]=0,接着就是遍历顺序了,但我们需要求出在这个字符串数组中0和1的个数,这就要先遍历数组了,之后在遍历是由于是01背包问题,要先遍历物品也就是str数组,在遍历背包也就是mn,重点是从前向后遍历,dp[i][j] 可以由前一个strs里的字符串推导出来,strs里的字符串有zeroNum个0,oneNum个1。dp[i][j] 就可以是 dp[i - zeroNum][j - oneNum] + 1。
然后我们在遍历的过程中,取dp[i][j]的最大值。
所以递推公式:dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
此时大家可以回想一下01背包的递推公式:dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);
对比一下就会发现,字符串的zeroNum和oneNum相当于物品的重量(weight[i]),字符串本身的个数相当于物品的价值(value[i])。
:
func findMaxForm(strs []string, m int, n int) int {
dp:=make([][]int,m+1)
for i,_ := range dp{
dp[i]=make([]int,n+1)
}
//获取0和1的数量
for i:=0;i<len(strs);i++{
zeroNum,oneNum:=0,0
for _,v:=range strs[i]{
if v=='0'{
zeroNum++
}
}
oneNum=len(strs[i])-zeroNum
for j:=m;j>=zeroNum;j--{
for k:=n;k>=oneNum;k--{
dp[j][k]=max(dp[j][k],dp[j-zeroNum][k-oneNum]+1)
}
}
}
return dp[m][n]
}
func max(a,b int)int{
if a>b{
return a
}
return b
}