Leetcode474. 一和零(动态规划-01背包问题)

给你一个二进制字符串数组 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 。

解题思路：

对于这道题，我们可以将二进制字符串数组 strs中每个字符看成一个个待选的物品，m,n代表背包容量，只不过这里是2个背包。因此我们运用0-1背包的思路来解这道题，对于传统的0-1背包问题，我们既可以采用2维dp求解，同时也可采用1维dp求解；若采用2维dp来求解，思路简单，但步骤繁琐，对应这里需要3维数组显得更加麻烦。最后我们采用1维dp来求解。

class Solution(object):
    def findMaxForm(self, strs, m, n):
        """
        :type strs: List[str]
        :type m: int
        :type n: int
        :rtype: int
        """
        #0-1背包问题
        #dp[m][n]代表当子集中最多有m个0和n个1时，最大子集的大小，共m+1,n+1个元素，0表示不含有0或 
        者1
        dp = [[0 for _ in range(n+1)] for _ in range(m+1)]
        for now_str in strs:
            zeros = now_str.count('0')
            ones = now_str.count('1')
            for i in range(m,zeros-1,-1):
                for j in range(n,ones-1,-1):
                    #取或者不取
                    dp[i][j] = max(dp[i][j],dp[i-zeros][j-ones]+1)
        return dp[m][n]