[NEO解题报告]《Leetcode》474. 一和零

1. 题目信息

1.1 题目描述

题目链接： 474. 一和零

给你一个二进制字符串数组 strs 和两个整数 m 和 n 。
请你找出并返回 strs 的最大子集的大小，该子集中 最多 有 m 个 0 和 n 个 1 。
如果 x 的所有元素也是 y 的元素，集合 x 是集合 y 的 子集 。

1.2 测试用例

• 示例 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

2. 题目分析

2.1 当做01背包问题+状态压缩

1. 详细的01背包问题， 建议阅读： 「代码随想录」474. 一和零【01背包】详解
2. 关于数组降为成滚动数组， 建议阅读： 动态规划：关于01背包问题，你该了解这些！（滚动数组）

• 根据题意， 最终结果限制 m 个 0 和 n 个 1, 所以可以用这两个信息来做多维dp， 定义为： dp[row][[m+1][n+1], 滚动数组方法压缩（去除第一维字符串的信息）后 dp[m+1][n+1];
• 那么最终输出的结果是 dp[m][n] ;
• 初始化 全 0 即可， 没有0 和 没有1的子集个数明显为0；
• 递推关系：
  • 假设第一维用row表示字符串的下标， 前面row-1个字符串组成的最大子集信息已经计算出来， 且当前第 row 行有 zeros个0 和 ones 个 1 ；
  • 则当前行的i个0和j个1的子集数为(选择当前行或者不选择当前行两种取最大)： dp[row][i][j] = max(dp[row-1][i][j], dp[row-1][i-zeros][j-ones]+1) ；
  • 压缩后的去除第一维递推关系为： dp[i][j] = max(dp[i][j], dp[i - zeros][j - ones] + 1) ， 注意遍历时需要从后往前遍历， dp数组中存储的原始值为上一行得到的结果， 从后向前计算保证不会重复使用和覆盖;

3. 代码详情

3.1 C++

3.1.1 当做01背包问题+状态压缩

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        // 执行用时：248 ms, 在所有 C++ 提交中击败了71.63%的用户
        // 内存消耗：9.4 MB, 在所有 C++ 提交中击败了77.02%的用户
        vector<vector<int>> dp(m + 1, vector<int>(n + 1));
        for (auto &s : strs) {
            int ones = 0;
            for (auto c: s) {
                if (c == '1') {
                    ones++;
                }
            }
            int zeros = s.length() - ones;

            for (int i = m; i >= zeros; i--) {
                for (int j = n; j >= ones; j--) {
                    dp[i][j] = max(dp[i][j], dp[i - zeros][j - ones] + 1);
                }
            }
        }
        return dp[m][n];
    }
};

3.2 Python

• 后续补充

4. 系列文章

• [NEO解题报告]《Leetcode》00 - 汇总