[leetCode]474. 一和零

题目

链接：https://leetcode-cn.com/problems/ones-and-zeroes

给你一个二进制字符串数组 strs 和两个整数 m 和 n 。

请你找出并返回 strs 的最大子集的大小，该子集中 最多 有 m 个 0 和 n 个 1 。

如果 x 的所有元素也是 y 的元素，集合 x 是集合 y 的 子集 。

动态规化

这题相当于问：把N个01字符串拆分放入0容量为m，1容量为n的背包中，能放的字符串的最大数量。

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        if (strs == null || strs.length == 0)
            return 0;
        // 字符串(物品)数量
        int N = strs.length;
        // dp[i][j][k], 使用前i个物品，背包容量能容纳j个0,k个1时，能容纳最多的字符串个数
        int[][][] dp = new int[N + 1][m + 1][n + 1];

        for (int i = 1; i <= N; i++) {
            int[] cnt = count(strs[i - 1]);
            for (int j = 0; j <= m; j++) {
                for (int k = 0; k <= n; k++) {
                    if (cnt[0] <= j && cnt[1] <= k) {// 装的下:装 || 不装
                        dp[i][j][k] = Math.max(dp[i-1][j][k], dp[i - 1][j - cnt[0]][k - cnt[1]] + 1);
                    } else {// 装不下
                        dp[i][j][k] = dp[i - 1][j][k];
                    }
                }
            }
        }
        return dp[N][m][n];
    }

    // 记录字符串中0和1的个数
    private int[] count(String str) {
        int[] cnt = {0,0};
        for (char c : str.toCharArray()) {
            cnt[c - '0']++;
        }
        return cnt;
    }
}

优化空间

通过观察可以发现当前状态只跟上一层左上侧有关

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        if (strs == null || strs.length == 0)
            return 0;
        // 字符串(物品)数量
        int N = strs.length;
        // dp[j][k], 使用前i个物品，背包容量能容纳j个0,k个1时，能容纳最多的字符串个数
        int[][] dp = new int[m + 1][n + 1];

        for (int i = 1; i <= N; i++) {
            int[] cnt = count(strs[i - 1]);
            for (int j = m; j >= 0; j--) {
                for (int k = n; k >= 0; k--) {
                    if (cnt[0] <= j && cnt[1] <= k) {// 装的下:装 || 不装
                        dp[j][k] = Math.max(dp[j][k], dp[j - cnt[0]][k - cnt[1]] + 1);
                    }
                }
            }
        }
        return dp[m][n];
    }

    // 记录字符串中0和1的个数
    private int[] count(String str) {
        int[] cnt = {0,0};
        for (char c : str.toCharArray()) {
            cnt[c - '0']++;
        }
        return cnt;
    }
}