leetcode 474. 一和零

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

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

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

注意:

  1. 给定 0 和 1 的数量都不会超过 100
  2. 给定字符串数组的长度不会超过 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" 。
class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        // int  dp[strs.size()+1][m+1][n+1]={0};三维数组初始化数据出现溢出,但不知为什么
          vector<vector<vector<int>>> dp(strs.size() + 1, vector<vector<int>>
                                      (m + 1 , vector<int>(n + 1 , 0)));

        for(int k=1;k<=strs.size();k++){
            int  cnt0=num0(strs[k-1]);
            int   cnt1=strs[k-1].size()-cnt0;
            // printf("第%d个字符串中'0'的数量为%d,'1'的数量为%d\n",k,cnt0,cnt1);
            

            for(int i=0;i<=m;i++)
            for(int j=0;j<=n;j++){
                // printf("K:%d,I:%d,J:%d\n",k,i,j);
                dp[k][i][j]=dp[k-1][i][j];
                if(i>=cnt0 && j>=cnt1)
                    dp[k][i][j]=max(dp[k-1][i][j],1+dp[k-1][i-cnt0][j-cnt1]);
            }
        }
        return dp[strs.size()][m][n];
    }
    int num0(string strs){
        int num=0;
        for(int i=0;i<strs.size();i++)
            if(strs[i]=='0')
                num++;
        return num;

    }
};
class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        // int  dp[strs.size()+1][m+1][n+1]={0};三维数组初始化数据出现溢出,但不知为什么
          vector<vector<int>> dp(m + 1 , vector<int>(n + 1 , 0));

        for(int k=1;k<=strs.size();k++){
            int  cnt0=num0(strs[k-1]);
            int   cnt1=strs[k-1].size()-cnt0;
            // printf("第%d个字符串中'0'的数量为%d,'1'的数量为%d\n",k,cnt0,cnt1);
            

            for(int i=m;i>=0;i--)
            for(int j=n;j>=0;j--){
            
                if(i>=cnt0 && j>=cnt1)
                    dp[i][j]=max(dp[i][j],1+dp[i-cnt0][j-cnt1]);
            }
        }
        return dp[m][n];
    }
    int num0(string strs){
        int num=0;
        for(int i=0;i<strs.size();i++)
            if(strs[i]=='0')
                num++;
        return num;

    }
};
发布了239 篇原创文章 · 获赞 16 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/qq_38662930/article/details/104081298