leetcode 474. 一和零

问题描述

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

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

提示:

1 <= strs.length <= 600
1 <= strs[i].length <= 100
strs[i] 仅由 '0' 和 '1' 组成
1 <= m, n <= 100

代码1

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        int size = strs.size(),k = 1;
        vector<vector<vector<int>>> dp(size+1,vector<vector<int>>(m+1,vector<int>(n+1,0)));
        for(string &s:strs)
        {
            int zero = 0,one = 0;
            for(char &c:s)
                c == '0'?++zero:++one;
            for(int i = 0; i <= m; ++i)
            {
                for(int j = 0; j <= n; ++j)
                {
                    if(i >= zero && j >= one)
                    {
                        dp[k][i][j] = max(dp[k-1][i][j],dp[k-1][i-zero][j-one]+1);
                    }
                    else
                        dp[k][i][j] = dp[k-1][i][j];
                }
            }
            ++k;
        }
        return dp[size][m][n];
    }
};

结果:

执行用时:600 ms, 在所有 C++ 提交中击败了14.11%的用户
内存消耗:100.3 MB, 在所有 C++ 提交中击败了7.97%的用户

代码2

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        int size = strs.size(),k = 1;
        vector<vector<int>> dp(m+1,vector<int>(n+1,0));
        for(string &s:strs)
        {
            int zero = 0,one = 0;
            for(char &c:s)
                c == '0'?++zero:++one;
            //为防止覆盖信息,倒着向前遍历
            for(int i = m; i >= zero; --i)
            {
                for(int j = n; j >= one; --j)
                {
                    dp[i][j] = max(dp[i][j],dp[i-zero][j-one]+1);
                }
            }
            ++k;
        }
        return dp[m][n];
    }
};

结果:

执行用时:260 ms, 在所有 C++ 提交中击败了77.96%的用户
内存消耗:9.7 MB, 在所有 C++ 提交中击败了64.64%的用户
posted @ 2020-09-30 15:00  曲径通霄  阅读(101)  评论(0编辑  收藏  举报