题目:
给你一个下标从 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.dfs查找所有可能被选择的列
2.遍历所有可能被选择的列, 然后遍历每一行检查这一行中的1是否在被选择的列里->结果ans+1
class Solution:
def maximumRows(self, matrix: List[List[int]], numSelect: int) -> int:
m, n = len(matrix), len(matrix[0])
s = []
def dfs(temp, j):
if len(temp) == numSelect:
s.append(temp[:])
return
for i in range(j, n):
temp.append(i)
dfs(temp, i + 1)
temp.pop()
dfs([], 0)
def isvalid(temp):
cnt = 0
for i in range(m):
t = False
for j in range(n):
if matrix[i][j] == 1 and j not in temp:
t = True
break
if not t:
cnt += 1
return cnt
ans = 0
for temp in s:
ans = max(ans, isvalid(temp))
return ans