leetcode 474.一和零

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

题解

朴素的dp
* 注意点在于对于每个字符串打表时应从打到小遍历，因为判断的过程中不可避免要判断下标更小的数组，而数组代表的是消耗ij的情况下是否选取当前字符串的最大利益，下标更小的数组则代表此前的状态
* 二维数组dp的两个下标代表了当前消耗i个0j个1的状态

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        int dp[m+5][n+5];
        memset(dp,0,sizeof(dp));
        vector<int> cost_one;
        vector<int> cost_zero;
        for(int i=0;i<strs.size();i++){
            int oneCount=0,zeroCount=0;
            for(int j=0;j<strs[i].size();j++){
                if(strs[i][j]=='0')
                    zeroCount++;
                else
                    oneCount++;
            }
            cost_one.push_back(oneCount);
            cost_zero.push_back(zeroCount);
        }
        for(int k=0;k<strs.size();k++){//对每个字符串进行判断
            //cout<<cost_one[k]<<' '<<cost_zero[k]<<endl;
            for(int i=m;i>=0;i--){
                for(int j=n;j>=0;j--){
                    if(i>=cost_zero[k] && j>=cost_one[k]){
                        dp[i][j] = max(1 + dp[i-cost_zero[k]][j-cost_one[k]],dp[i][j]);
                    }
                }
            }
        }
        return dp[m][n];
    }
};

/*
    设dp[i][j]为i个0，j个1所能获得的最大价值
    考虑状态转移方程：
        每转移一次都对ij重新打表，易发现遍历条件为:
            if(i >= cost_zero[str] && j >= cost_one[str])
        dp[i][j] = max(1+dp[i-cost_zero[k]][j-cost_one[j]],dp[i][j]);    //选或不选 言下之意就是从前往后对于每个字符串，判断选与不选时的最大价值
 

*/