LeetCode 474 Ones and Zeroes 题解

题意简述：有若干个由0和1组成的字符串，现在限定使用m个0和n个1，每个0和1至多使用1次，问最多能组成多少个给定的字符串。
输入：字符串的vector，m和n的值
输出：能组成的字符串的数目
示例：给出m=5，n=3。字符串有{“10”, “0001”, “111001”, “1”, “0”}，最多可以组成4个字符串：“10,”0001”,”1”,”0”。

---

题解：
如果把提供的0和1看作容量分别为m和n的背包，把字符串的0和1看作对应需要的容量（例如字符串0001需要3个容量0和1个容量1），那么这个问题就可以看作二维的01背包问题。
跟一维的01背包的空间优化相同，二维的01背包可以只使用二维数组存放中间结果。考虑对于第i个字符串，0容量为j，1容量为k的情况下最多组成的字符串数，其状态转移方程如下，其中z[i]表示第i个字符串需要的0容量，o[i]表示第i个字符串需要的1容量。

$$dp(j,k)=max(dp(j,k),dp(j-z[i],k-o[i]+1)\ j=m,m-1,...,z[i], k=n,n-1,...,o[i]$$

算法的时间复杂度是$O(n^3)$，空间复杂度是$O(n^2)$。

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        if(strs.size() == 0) return 0;

        vector<int> ones,zeros;
        for(int i = 0;i < strs.size();i++) {
            bitset<200> temp(strs[i]);
            ones.push_back(temp.count());
            zeros.push_back(strs[i].size()-temp.count());
        }

        vector<vector<int>> dp;
        for(int i = 0;i <= m;i++) {
            vector<int> temp(n+1,0);
            dp.push_back(temp);
        }

        for(int i = 0;i < strs.size();i++) {
            for(int j = m;j >= zeros[i];j--) {
                for(int k = n;k >= ones[i];k--) {
                    dp[j][k] = max(dp[j][k], dp[j-zeros[i]][k-ones[i]]+1);
                }
            }
        }

        return dp[m][n];
    }
};