Leetcode 474. 一和零

在计算机界中，我们总是追求用有限的资源获取最大的收益。

现在，假设你分别支配着 m 个 0 和 n 个 1。另外，还有一个仅包含 0 和 1 字符串的数组。

你的任务是使用给定的 m 个 0 和 n 个 1 ，找到能拼出存在于数组中的字符串的最大数量。每个 0 和 1 至多被使用一次。

注意:

给定 0 和 1 的数量都不会超过 100。
给定字符串数组的长度不会超过 600。

思路：

01背包的变换问题，不同的是在背包问题中，我们只有一种容量，而在这道题中，我们有 0 和 1 两种容量。每个物品（字符串）需要分别占用 0 和 1 的若干容量，并且所有物品的价值均为 1。

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        int size = strs.length;
        int z = m + 1;//zero
        int o = n + 1;//one
        int[][][] dp = new int[size][z][o];
        int[] cnt = getZerosAndOnes(strs[0]);
        for (int i = 0; i < z; i++) {
            for (int j = 0; j < o; j++) {
                if (i >= cnt[0] && j >= cnt[1])
                    dp[0][i][j] = 1;
            }
        }
        for (int s = 1; s < size; s++) {
            for (int i = 0; i < z; i++) {
                for (int j = 0; j < o; j++) {
                    cnt = getZerosAndOnes(strs[s]);
                    if (i - cnt[0] >= 0 && j - cnt[1] >= 0)
                        dp[s][i][j] = Math.max(dp[s - 1][i][j], 1+dp[s - 1][i - cnt[0]][j - cnt[1]]);
                    else
                        dp[s][i][j] = dp[s-1][i][j];
                }
            }
        }
        return dp[size - 1][m][n];
    }

    private int[] getZerosAndOnes(String str) {
        int[] res = new int[2];
        for (int i = 0; i < str.length(); i++) {
            if (str.charAt(i) == '1')
                res[1]++;
            else
                res[0]++;
        }
        return res;
    }
}