每日力扣-01背包(动态规划)-01个数

214 阅读1分钟

给你一个二进制字符串数组 strs 和两个整数 m 和 n 。

请你找出并返回 strs 的最大子集的大小,该子集中 最多 有 m 个 0 和 n 个 1 。

如果 x 的所有元素也是 y 的元素,集合 x 是集合 y 的 子集 。

输入:strs = ["10", "0001", "111001", "1", "0"], m = 5, n = 3 输出:4 解释:最多有 5 个 0 和 3 个 1 的最大子集是 {"10","0001","1","0"} ,因此答案是 4 。 其他满足题意但较小的子集包括 {"0001","1"} 和 {"10","1","0"} 。{"111001"} 不满足题意,因为它含 4 个 1 ,大于 n 的值 3 。

输入:strs = ["10", "0", "1"], m = 1, n = 1 输出:2 解释:最大的子集是 {"0", "1"} ,所以答案是 2 。

解题思路:对于第i个字符串,可以选择或者不选择,要是第i个字符串中的0或1个数大于背包容量,那肯定不选择,如果小于背包容量,那就再看装了和不装哪个字符串多。 比如说,如果你想把一个cnt = [1,2]的字符串装进背包(在容量足够的前提下),只需要找到容量为[j - 1][k - 2]时候的字符串数再加上1,就可以得到装入后的字符串数了。

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        int len = strs.size();
        vector<vector<int>> dp(m+1,vector<int>(n+1,0));
        for(int i = 0 ; i < len ; i++)
        {
            vector<int> cnt = count(strs[i]);
            for(int j = m ; j >= 0 ; j--)
            {
                for(int k = n ; k >= 0 ; k--)
                {
                    if(cnt[0] > j || cnt[1] > k)
                    {
                        dp[j][k] = dp[j][k];
                    }
                    else
                    {
                        dp[j][k] = max(dp[j-cnt[0]][k-cnt[1]] + 1 , dp[j][k]);
                    }
                }
            }
        }
        return dp[m][n];
    }
    vector<int> count(string str)
    {
        vector<int> cnt(2,0);
        int n = str.size();
        for(int i = 0 ; i < n ; i++)
        {
            cnt[str[i] - '0']++;
        }
        return cnt;
    }
};