leetcode474. 一和零

传送门

题目：假设你分别支配着 m 个 0 和 n 个 1。另外，还有一个仅包含 0 和 1 字符串的数组。你的任务是使用给定的 m 个 0 和 n 个 1 ，找到能拼出存在于数组中的字符串的最大数量。每个 0 和 1 至多被使用一次。

输入: Array = {“10”, “0001”, “111001”, “1”, “0”}, m = 5, n = 3
输出: 4
解释: 总共 4 个字符串可以通过 5 个 0 和 3 个 1 拼出，即 “10”,“0001”,“1”,“0” 。

先熟悉一下 0-1背包

每一个字符串视为装进背包的物品。这道题就可以使用 0-1 背包问题的思路完成。这里的目标值是能放进背包的字符串的数量。思路依然是“一个一个尝试，容量一点一点尝试”，每个物品分类讨论：选与不选。

第 1 步：定义状态

依然是尝试「题目问啥，就把啥定义成状态」。
dp[i][j][k] 表示输入字符串在子区间 [0, i] 能够使用 j 个 0 和 k 个 1 的字符串的最大数量。

第 2 步：状态转移方程 注意 = 右边都是i-1

dp[i][j][k]= 
dp[i - 1][j][k], 不选当前考虑的字符串，至少是这个数值
dp[i - 1][j - 当前字符串使用0的个数][k - 当前字符串使用1的个数] + 1  选择当前考虑的字符串

	public int findMaxForm(String[] strs, int m, int n) {
        if (m == 0 && n == 0) return 0;
        int[][][] dp = new int[strs.length + 1][m + 1][n + 1];   
       
        //i从0开始的话，下面有i-1，要特判，m和n本来就要+1因为m和n值在0 到m(n) 之间取
        for (int i = 1; i <= strs.length; ++i) {
            int[] cnt = countZerosAndOnes(strs[i - 1]);
			// 不能从cnt[0]开始，因为j<cnt[0]的值 要更新成上一行的
            for (int j = 0; j <= m; ++j) { 
                for (int k = 0; k <= n; ++k) {
                    if (j >= cnt[0] && k >= cnt[1]) 
                        dp[i][j][k] =
                            Math.max(dp[i - 1][j][k], 
                            dp[i - 1][j - cnt[0]][k - cnt[1]] + 1);// 这里是i-1不是i
                      else
							dp[i][j][k] = dp[i - 1][j][k];// 选不了剩下的字符不够
                }
            }
        }
        return dp[strs.length][m][n];
    }
    int[] countZerosAndOnes(String str) {// 统计字符串str中的0，1的个数
        int[] cnt = new int[2];
        for (char ch : str.toCharArray()) {
            cnt[ch - '0']++;
        }
        return cnt;
    }