LeetCode - 474. 一和零

474. 一和零

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

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

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

注意:

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

解题思路： 此题考查01背包问题，01背包问题的特征是对于数组中每个元素有选和不选两个选择。接下来按照01背包问题的模板进行解题，dp[i][j][k]表示前i个字符串在消耗j个0和k个1时最多能拼接出的字符串个数，那么对于第i-1个字符串有选和不选两种情况，如果选的话，则能拼接出dp[i - 1][j - a_0][k - a_1] + 1个字符串，如果不选的话，则能拼接出dp[i - 1][j][k]个字符串，在两种情况下取最大值即可。

背包问题九讲

// 01背包问题
// dp[i][j][k]表示前i个字符串在消耗j个0和k个1时最多能拼接出的字符串个数
// dp[i][j][k] = max(dp[i - 1][j][k], dp[i - 1][j - a_0][k - a_1] + 1)
// Time: O(size * m * n)
// Space: O(size * m * n)
class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        int size = strs.size();
        vector<int> ones, zeros;
        for (auto str : strs) {
            int a = 0, b = 0;
            for (auto c : str) {
                if (c == '1') ++a;
                else if (c == '0') ++b;
            }
            ones.push_back(a);
            zeros.push_back(b);
        }
        vector<vector<vector<int>>> dp(size + 1, vector<vector<int>>(m + 1, vector<int>(n + 1, 0)));
        for (int i = 1; i <= size; ++i) {
            for (int j = 0; j <= m; ++j) {
                for (int k = 0; k <= n; ++k) {
                    dp[i][j][k] = max(dp[i][j][k], dp[i - 1][j][k]);
                    if (j >= zeros[i - 1] && k >= ones[i - 1]) {
                        dp[i][j][k] = max(dp[i][j][k], dp[i - 1][j - zeros[i - 1]][k - ones[i - 1]] + 1);
                    }
                }
            }
        }
        return dp[size][m][n];
    }
};