474. 一和零

474. 一和零

1.题目描述

在计算机界中，我们总是追求用有限的资源获取最大的收益。现在，假设你分别支配着 m 个 0 和 n 个 1。另外，还有一个仅包含 0 和 1 字符串的数组。你的任务是使用给定的 m 个 0 和 n 个 1 ，找到能拼出存在于数组中的字符串的最大数量。每个 0 和 1 至多被使用一次。
注意:
1.给定 0 和 1 的数量都不会超过 100。
2.给定字符串数组的长度不会超过 600。
示例 1:

示例 2:

2.思路（动态规划）

这是一个多维费用的 0-1 背包问题，有两个背包大小，0 的数量和 1 的数量。
（1）定义状态：
dp[k][i][j] 为在区间[0,k]内使用i个0和j和1拼接出字符串的最大数量
（2）状态转移方程：
1.不选择当前字符串
2.选择当前字符串，前一个状态下用i-zeros个0和j-ones个1得到的最大数量+1
$$dp[k][i][j]=\{\begin{array}{ll}dp[k-1][i][j],& \\ dp[k-1][i-zeros][j-ones]+1& \end{array}$$
（3）初始化：
为了避免分类讨论，通常多设置一行。这里可以认为，第 0 个字符串是空串。第 1 行默认初始化为 0。
（4）输出：
输出是最后一个状态，即：dp[len][m][n]。
空间压缩：
因为只与上一状态有关，因此可以去掉维度k。

3.代码

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        if(strs.empty()){
            return 0;
        }
        vector<vector<int>> dp(m+1,vector<int>(n+1,0));
        for(auto &str : strs){
            int ones = 0, zeros = 0;//统计每个字符串0和1的个数
            for(auto &s : str){
                if(s == '0'){
                    zeros++;
                }
                else if(s == '1'){
                    ones++;
                }
            }
            for(int i = m;i >= zeros;--i){
                for(int j = n;j >= ones;--j){
                    dp[i][j] = max(dp[i][j],dp[i-zeros][j-ones] + 1);//选择当前字符串+1，不选择当前字符串
                }
            }
        }
        return dp[m][n];
    }
};

4.复杂度分析

时间复杂度：O(mnl)，l是字符串的数量
空间复杂度：O(mn)