ones-and-zeroes.md

474. 一和零

给你一个二进制字符串数组 strs 和两个整数 m 和 n 。

请你找出并返回 strs 的最大子集的长度，该子集中 最多 有 m 个 0 和 n 个 1 。

如果 x 的所有元素也是 y 的元素，集合 x 是集合 y 的 子集 。

示例 1：

输入：strs = ["10", "0001", "111001", "1", "0"], m = 5, n = 3
输出：4
解释：最多有 5 个 0 和 3 个 1 的最大子集是 {"10","0001","1","0"} ，因此答案是 4 。
其他满足题意但较小的子集包括 {"0001","1"} 和 {"10","1","0"} 。{"111001"} 不满足题意，因为它含 4 个 1 ，大于 n 的值 3 。

示例 2：

输入：strs = ["10", "0", "1"], m = 1, n = 1
输出：2
解释：最大的子集是 {"0", "1"} ，所以答案是 2 。

提示：

• 1 <= strs.length <= 600
• 1 <= strs[i].length <= 100
• strs[i] 仅由 '0' 和 '1' 组成
• 1 <= m, n <= 100

题解：

impl Solution {
    pub fn find_max_form(strs: Vec<String>, m: i32, n: i32) -> i32 {
        let len = strs.len();
        let m = m as usize;
        let n = n as usize;
        let mut dp = vec![vec![vec![0;n+1];m+1];len+1];

        for i in 1..=len {
            let (zeros, ones) = Self::count_zeros(strs[i-1].as_str());

            for j in 0..=m {
                for k in 0..=n {
                    dp[i][j][k] = dp[i-1][j][k];

                    if j >= zeros && k >= ones {
                        dp[i][j][k] = (dp[i][j][k]).max(dp[i-1][j-zeros][k-ones] + 1);
                    }
                }
            }
        }

        dp[len][m][n] as i32
    }

    fn count_zeros(s: &str) -> (usize, usize) {
        let cnt = s.chars().filter(|&c| c == '0').count();

        (cnt, s.len() as usize - cnt)
    }
}