474. 一和零：动态规划+回溯

https://leetcode.cn/problems/ones-and-zeroes/

题目要求

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

回溯

class Solution {
    public int MaxSize = 0;
    HashMap<String, int[]> map = new HashMap<>();

    public int findMaxForm(String[] strs, int m, int n) {
        for (String str : strs) {
            int num0 = 0, num1 = 0;
            for (int i = 0; i < str.length(); i++) {
                if (str.charAt(i) == '1') num1++;
                else num0++;
            }
            map.put(str, new int[]{num0, num1});
        }
        backtrack(strs, m, n, 0, 0);
        return MaxSize;
    }

    public void backtrack(String[] strs, int m, int n, int cur, int index) {
        if (m < 0 || n < 0)
            return;
        MaxSize = Math.max(cur, MaxSize);
        for (int i = index; i < strs.length; i++) {
            backtrack(strs, m - map.get(strs[i])[0], n - map.get(strs[i])[1], cur + 1, i + 1);
        }
    }
}

• 回溯解法超时

动态规划

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        int[][] dp = new int[m + 1][n + 1];
        for (String str : strs) {
            int num_0 = 0, num_1 = 0;
            for (int i = 0; i < str.length(); i++) {
                if (str.charAt(i) == '0')
                    num_0++;
                else
                    num_1++;
            }
            // 倒序遍历，否则重复统计
            for (int i = m; i >= num_0; i--) {
                for (int j = n; j >= num_1; j--) {
                    dp[i][j] = Math.max(dp[i][j], dp[i - num_0][j - num_1] + 1);
                }
            }
        }
        return dp[m][n];
    }
}

• dp[i][j]：最多有i个0和j个1的strs的最大子集的大小为dp[i][j]。
• 递推公式：dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
• 内层for循环遍历背包容量且从后向前遍历，否则会重复统计同一个物品，相当于一个物品被放了两次；