Leetcode 474. 一和零 C++

Leetcode 474. 一和零

题目

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

现在，假设你分别支配着 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" 。

题解

动态规划
令dp[i][j][k]表示，前i个数中有j个0、k个1供使用所能拼出的最大收益
假设strs[i]需要用zero个0、one个1

• j<zero || k<one，说明我们的0、1不够用来拼strs[i]，故dp[i][j][k] = dp[i-1][j][k];

• 其他情况下，我们可以选择拼这个字符串，即dp[i-1][j-zero][k-one]+1；也可以选择不拼，即dp[i-1][j][k]。故dp[i][j][k] = max(dp[i-1][j][k],dp[i-1][j-zero][k-one]+1)
  考虑一下初始化，对于strs[0]，也就是dp[0][j][k]

• j<zero || k<one，说明不够拼，故dp[0][j][k] = 0

• 其他情况下，我们进行拼就是最大收益，即dp[0][j][k] = 1

详细过程见代码

代码

int findMaxForm(vector<string>& strs, int m, int n) {
        int size = strs.size();
        if(size == 0)   return 0;
        vector<vector<vector<int>>> dp(size,vector<vector<int>>(m+1,vector<int>(n+1,0)));
        int zero=0,one=0;
        for(int i=0; i<strs[0].size(); i++){
            if(strs[0][i]=='1')    one++;
            else    zero++;
        }
        for(int j=0; j<=m; j++){
            for(int k=0; k<=n; k++){
                if(j<zero || k<one)
                    dp[0][j][k] = 0;
                else
                    dp[0][j][k] = 1;
            }
        }
        for(int i=1; i<size; i++){
            zero = 0;
            one = 0;
            for(int j=0; j<strs[i].length(); j++){
                if(strs[i][j]=='1')    one++;
                else    zero++;
            }
            for(int j=0; j<=m; j++){
                for(int k=0; k<=n; k++){
                    if(j<zero || k<one)
                        dp[i][j][k] = dp[i-1][j][k];
                    else
                        dp[i][j][k] = max(dp[i-1][j][k],dp[i-1][j-zero][k-one]+1);
                }
            }
        }
        return dp[size-1][m][n];
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ones-and-zeroes
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。