【LeetCode】474. 一和零 （动态规划）

1. 问题描述

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

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

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

示例 1:

输入: strs = ["10", "0001", "111001", "1", "0"], m = 5, n = 3
输出: 4
解释: 总共 4 个字符串可以通过 5 个 0 和 3 个 1 拼出，即 "10","0001","1","0" 。
示例 2:
输入: strs = ["10", "0", "1"], m = 1, n = 1
输出: 2
解释: 你可以拼出 "10"，但之后就没有剩余数字了。更好的选择是拼出 "0" 和 "1" 。

2. 思路

该题是典型的动态规划问题，首先构建状态转移方程。

dp[i][j] 表示i个0，j个1时，能拼出存在于数组中的字符串的最大数量。

zeroNum 表示 当前字符串中0的个数
oneNum 表示 当前字符串中1的个数

dp[i][j] = max(dp[i][j], 1 + dp[i-zeroNum][j-oneNum])

3. 代码

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        if (strs.length == 0) {
            return 0;
        }

        int[][] dp = new int[m+1][n+1];
        for (String s : strs) {
            int zeroNum = 0;
            int oneNum = 0;
            for (char c : s.toCharArray()){
                if (c == '0') {
                    zeroNum++;
                } else {
                    oneNum++;
                }
            }

            for(int i = m; i >=zeroNum;i--) {
                for (int j = n; j >=oneNum; j--) {
                    dp[i][j] = Math.max(dp[i][j], dp[i-zeroNum][j-oneNum] + 1);
                }
            }
        }
        return dp[m][n];
    }
}