def dfs(matrix, row, col, visited):
"""
递归 DFS 函数,检查矩阵中指定位置是否符合要求
:param matrix: 二进制矩阵
:param row: 当前检查的行
:param col: 当前检查的列
:param visited: 已经访问过的位置
:return: 当前位置是否符合要求
"""
if row < 0 or col < 0 or row >= len(matrix) or col >= len(matrix[0]) or (row, col) in visited:
# 超出矩阵范围或已经访问过该位置,则返回 False
return False
visited.add((row, col)) # 将当前位置标记已访问
# 检查当前位置是否符合要求
if matrix[row][col] == 1:
for r in range(len(matrix)):
if r != row and matrix[r][col] != 0:
# 检查所在列是否符合要求
return False
for c in range(len(matrix[0])):
if c != col and matrix[row][c] != 0:
# 检查所在行是否符合要求
return False
return True
else:
return False
def find_ones(matrix):
"""
查找二进制矩阵中所有符合要求的位置
:param matrix: 二进制矩阵
:return: 符合要求的位置列表
"""
ones = []
visited = set() # 已经访问过的位置
for i in range(len(matrix)):
for j in range(len(matrix[0])):
if dfs(matrix, i, j, visited):
ones.append((i, j))
return ones
这里的解决方法采用了深度优先搜索(DFS)算法,用一个辅助函数 dfs 来递归检查当前位置是否符合要求。具体步骤如下: