Medium

Ones and ZeroesC++

Full explanation · Time O(s * m * n) · Space O(m * n)

// Time:  O(s * m * n), s is the size of the array.
// Space: O(m * n)

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        vector<vector<int>> dp(m + 1, vector<int>(n + 1));
        for (const auto &str : strs) {
            int zero_count = 0, one_count = 0;
            for (const auto& c : str) {
                if (c == '0') {
                    ++zero_count;
                } else if (c == '1') {
                    ++one_count;
                }
            }

            for (int i = m; i - zero_count >= 0; --i) {
            	for (int j = n; j - one_count >= 0; --j) {
                    dp[i][j] = max(dp[i][j], dp[i - zero_count][j - one_count] + 1);
            	}
            }
        }
        return dp[m][n];
    }
};