一.题目描述
在计算机界中,我们总是追求用有限的资源获取最大的收益。
现在,假设你分别支配着 m 个 0 和 n 个 1。另外,还有一个仅包含 0 和 1 字符串的数组。
你的任务是使用给定的 m 个 0 和 n 个 1 ,找到能拼出存在于数组中的字符串的最大数量。每个 0 和 1 至多被使用一次。
注意:
给定 0 和 1 的数量都不会超过 100。
给定字符串数组的长度不会超过 600。
示例 1:
输入: Array = {"10", "0001", "111001", "1", "0"}, m = 5, n = 3
输出: 4
解释: 总共 4 个字符串可以通过 5 个 0 和 3 个 1 拼出,即 "10","0001","1","0" 。
示例 2:
输入: Array = {"10", "0", "1"}, m = 1, n = 1
输出: 2
解释: 你可以拼出 "10",但之后就没有剩余数字了。更好的选择是拼出 "0" 和 "1" 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ones-and-zeroes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二.思路分析
这就是一个二维背包问题,使用动态规划求解
0和1的个数是限制条件,相当于背包问题中背包的容积,只不过这里有了两个约束条件
能放进去的字符串的个数相当于背包问题需要最大化的物品总价值
递推关系式为:
DP[k][i][j]=max{DP[k-1][i][j],DP[k-1][i-count0][j-count1]+1}
上面的DP[k][i][j]表示前k个物品,在i个0,j个1的约束下能放下的最大个数
DP[k][i][j]的求解分为两种情况考虑:包含第k个物品,这样就是前k-1个物品对应的在i-count0个0和j-count1个1下能放下的最大个数加1
不包含第i个物品,还是等于前k-1个物品在i个0j个1的约束下的最大个数
count0表示第k个物品中0的个数,count1表示第k个物品中1的个数
这种在约束下求最大值的问题一般都可以归为背包问题!!!
代码:对空间做了优化,将三维空间变成了二维存储空间
class Solution:
def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
DP=[[0 for i in range(n+1)] for j in range(m+1)]
for Str in strs:
C1=C0=0
for Char in Str:
if Char=="1":
C1+=1
else:
C0+=1
for i in range(m,C0-1,-1):
for j in range(n,C1-1,-1):
if DP[i-C0][j-C1]+1>DP[i][j]:
DP[i][j]=DP[i-C0][j-C1]+1
return DP[m][n]