被列覆盖的最多行数
难度:
标签:
题目描述
给你一个下标从 0 开始、大小为 m x n
的二进制矩阵 matrix
;另给你一个整数 numSelect
,表示你必须从 matrix
中选择的 不同 列的数量。
如果一行中所有的 1
都被你选中的列所覆盖,则认为这一行被 覆盖 了。
形式上,假设 s = {c1, c2, ...., cnumSelect}
是你选择的列的集合。对于矩阵中的某一行 row
,如果满足下述条件,则认为这一行被集合 s
覆盖:
- 对于满足
matrix[row][col] == 1
的每个单元格matrix[row][col]
(0 <= col <= n - 1
),col
均存在于s
中,或者 row
中 不存在 值为1
的单元格。
你需要从矩阵中选出 numSelect
个列,使集合覆盖的行数最大化。
返回一个整数,表示可以由 numSelect
列构成的集合 覆盖 的 最大行数 。
示例 1:
输入:matrix = [[0,0,0],[1,0,1],[0,1,1],[0,0,1]], numSelect = 2 输出:3 解释: 图示中显示了一种覆盖 3 行的可行办法。 选择 s = {0, 2} 。 - 第 0 行被覆盖,因为其中没有出现 1 。 - 第 1 行被覆盖,因为值为 1 的两列(即 0 和 2)均存在于 s 中。 - 第 2 行未被覆盖,因为 matrix[2][1] == 1 但是 1 未存在于 s 中。 - 第 3 行被覆盖,因为 matrix[2][2] == 1 且 2 存在于 s 中。 因此,可以覆盖 3 行。 另外 s = {1, 2} 也可以覆盖 3 行,但可以证明无法覆盖更多行。
示例 2:
输入:matrix = [[1],[0]], numSelect = 1 输出:2 解释: 选择唯一的一列,两行都被覆盖了,因为整个矩阵都被覆盖了。 所以我们返回 2 。
提示:
m == matrix.length
n == matrix[i].length
1 <= m, n <= 12
matrix[i][j]
要么是0
要么是1
1 <= numSelect <= n
代码结果
运行时间: 24 ms, 内存: 16.7 MB
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
// 思路:
// 1. 找出所有可能的列组合。
// 2. 遍历每个组合,检查其能覆盖的行数。
// 3. 返回能覆盖行数最多的列组合。
public class Solution {
public int maximumRows(int[][] matrix, int numSelect) {
int m = matrix.length;
int n = matrix[0].length;
List<List<Integer>> combinations = generateCombinations(n, numSelect);
return combinations.stream()
.mapToInt(combination -> (int) IntStream.range(0, m)
.filter(i -> isCovered(matrix[i], combination))
.count())
.max()
.orElse(0);
}
private boolean isCovered(int[] row, List<Integer> combination) {
return IntStream.range(0, row.length)
.allMatch(j -> row[j] == 0 || combination.contains(j));
}
private List<List<Integer>> generateCombinations(int n, int numSelect) {
List<List<Integer>> combinations = new ArrayList<>();
combine(combinations, new ArrayList<>(), 0, n, numSelect);
return combinations;
}
private void combine(List<List<Integer>> combinations, List<Integer> current, int start, int n, int numSelect) {
if (numSelect == 0) {
combinations.add(new ArrayList<>(current));
return;
}
for (int i = start; i < n; i++) {
current.add(i);
combine(combinations, current, i + 1, n, numSelect - 1);
current.remove(current.size() - 1);
}
}
}
解释
方法:
此题解使用了位运算和掩码的概念来解决问题。首先,将每行的1所在的列位置转换为二进制数的掩码,例如行[1,0,1]转换为掩码5(101 in binary)。之后,生成所有可能的列组合掩码,并检查每个组合是否由精确的'numSelect'位1。对于每个有效的列掩码,计算有多少行可以完全被这些列覆盖,即行掩码和列掩码的与操作结果等于行掩码本身。最后,找到可以覆盖最多行的列组合,并返回覆盖行数。
时间复杂度:
O(m * 2^n)
空间复杂度:
O(m)
代码细节讲解
🦆
在题解中,如何确保生成的列掩码包含恰好numSelect个1?
▷🦆
为什么题解中选择使用位掩码来表示列的组合,这种方法与其他可能的数据结构相比有何优势?
▷🦆
在计算行被覆盖的情况时,行掩码和列掩码的与操作结果等于行掩码本身意味着什么?
▷🦆
如何处理矩阵中空行(即没有1的行)?在算法中对这种特殊情况有没有进行特别的处理?
▷