https://leetcode.cn/problems/ones-and-zeroes/
题目要求
给你一个二进制字符串数组 strs 和两个整数 m 和 n 。请你找出并返回 strs 的最大子集的长度,该子集中 最多 有 m 个 0 和 n 个 1 。如果 x 的所有元素也是 y 的元素,集合 x 是集合 y 的 子集 。
回溯
class Solution {
public int MaxSize = 0;
HashMap<String, int[]> map = new HashMap<>();
public int findMaxForm(String[] strs, int m, int n) {
for (String str : strs) {
int num0 = 0, num1 = 0;
for (int i = 0; i < str.length(); i++) {
if (str.charAt(i) == '1') num1++;
else num0++;
}
map.put(str, new int[]{num0, num1});
}
backtrack(strs, m, n, 0, 0);
return MaxSize;
}
public void backtrack(String[] strs, int m, int n, int cur, int index) {
if (m < 0 || n < 0)
return;
MaxSize = Math.max(cur, MaxSize);
for (int i = index; i < strs.length; i++) {
backtrack(strs, m - map.get(strs[i])[0], n - map.get(strs[i])[1], cur + 1, i + 1);
}
}
}
- 回溯解法超时
动态规划
class Solution {
public int findMaxForm(String[] strs, int m, int n) {
int[][] dp = new int[m + 1][n + 1];
for (String str : strs) {
int num_0 = 0, num_1 = 0;
for (int i = 0; i < str.length(); i++) {
if (str.charAt(i) == '0')
num_0++;
else
num_1++;
}
// 倒序遍历,否则重复统计
for (int i = m; i >= num_0; i--) {
for (int j = n; j >= num_1; j--) {
dp[i][j] = Math.max(dp[i][j], dp[i - num_0][j - num_1] + 1);
}
}
}
return dp[m][n];
}
}
- dp[i][j]:最多有i个0和j个1的strs的最大子集的大小为dp[i][j]。
- 递推公式:
dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
- 内层for循环遍历背包容量且从后向前遍历,否则会重复统计同一个物品,相当于一个物品被放了两次;