18-今日一扣(LeetCode)-474-DP01背包-M

474. 一和零

LeetCode

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

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

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

注意:

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

示例 1:

输入: Array = {"10", "0001", "111001", "1", "0"}, m = 5, n = 3
输出: 4

解释: 总共 4 个字符串可以通过 5 个 0 和 3 个 1 拼出，即 "10","0001","1","0" 。

示例 2:

输入: Array = {"10", "0", "1"}, m = 1, n = 1
输出: 2

解释: 你可以拼出 "10"，但之后就没有剩余数字了。更好的选择是拼出 "0" 和 "1" 。

题解

本题是01背包问题的多维度扩展版本

01背包限制只有背包的容量。所以我们的dp数组第二维度是容量j。

这个问题对于每个元素同样与取和不取两种。但是限制不在是一个，而是两个，1不能超过m，0不能超过n。

那么我们的dp数组为$dp[i][j][k]$，记录前i个元素，在1的数量为j，0的数量为k下的最大收益。

我们的初始条件为$dp[\ast ][0][0]=0$即一个元素都不选，且1的数量、0的数量都为0时，我们的最大收益为0.
$$dp[i][j][k]=\max \{\begin{array}{l}dp[i-1][j-nums1][k-nums0]+1选择该字符串\\ dp[i-1][j][k]不选择该字符串\end{array}$$
在经过数组维度优化，可以将三维数组优化到二维数组：

最终方案：

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        int len = strs.length;
        int [][] dp = new int[m+1][n+1];
        //初始条件
        dp[0][0] = 0;
        
        for(int i = 0; i<strs.length; i++){
            int nums1 = count1(strs[i]);
            int nums0 = strs[i].length()-nums1;
            for(int j = m; j-nums0>=0; j--){
                for(int k = n; k-nums1>=0; k--){
                    dp[j][k] = Math.max(dp[j-nums0][k-nums1]+1, dp[j][k]);
                }
            }
        }
        return dp[m][n];
    }

    private int count1(String str){
        int ans = 0;
        for(int i=0; i<str.length(); i++){
            if(str.charAt(i)=='1'){
                ans++;
            }
        }
        return ans;
    }
}