题目
给你一个二进制字符串数组 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 。
代码
/*
class Solution {
public int findMaxForm(String[] strs, int m, int n) {
int len = strs.length;
// 预处理每一个字符包含 0 和 1 的数量
int[][] cnt = new int[len][2];
for (int i = 0; i < len; i++) {
String str = strs[i];
int zero = 0, one = 0;
for (char c : str.toCharArray()) {
if (c == '0') {
zero++;
} else {
one++;
}
}
cnt[i] = new int[]{zero, one};
}
int[][][]f=new int[len+1][m+1][n+1];
// 处理考虑其余物品的情况
for (int k = 1; k <=len; k++) {
int zero = cnt[k-1][0], one = cnt[k-1][1];
for (int i = 0; i <= m; i++) {
for (int j = 0; j <= n; j++) {
// 不选择第 k 件物品
int a = f[k-1][i][j];
// 选择第 k 件物品(前提是有足够的 m 和 n 额度可使用)
int b = (i >= zero && j >= one) ? f[k-1][i-zero][j-one] + 1 : 0;
f[k][i][j] = Math.max(a, b);
}
}
}
return f[len][m][n];
}
}
*/
class Solution {
public int findMaxForm(String[] strs, int m, int n) {
int len = strs.length;
// 预处理每一个字符包含 0 和 1 的数量
int[][] cnt = new int[len][2];
for (int i = 0; i < len; i++) {
String str = strs[i];
int zero = 0, one = 0;
for (char c : str.toCharArray()) {
if (c == '0') {
zero++;
} else {
one++;
}
}
cnt[i] = new int[]{zero, one};
}
int[][]f=new int[m+1][n+1];
// 处理考虑其余物品的情况
for (int k = 1; k <=len; k++) {
int zero = cnt[k-1][0], one = cnt[k-1][1];
for (int i = m; i >= 0; i--) {
for (int j = n; j >= 0; j--) {
// 不选择第 k 件物品
int a = f[i][j];
// 选择第 k 件物品(前提是有足够的 m 和 n 额度可使用)
int b = (i >= zero && j >= one) ? f[i-zero][j-one] + 1 : 0;
f[i][j] = Math.max(a, b);
}
}
}
return f[m][n];
}
}
反思
1、三维的,简化空间的方法和二维的相识,反转去除第一维即可