![](/img/trans.png)
[英]How can I get 3D coordinates of a 3D matrix consisting of only 0s and 1s
[英]2D matrix containing only 0s and 1s - python
给定的二维矩阵只包含 0 和 1。 每个 0 代表土地,每个 1 代表河流的一部分。 一条河流由任意数量的 1 组成,这些 1 水平或垂直相邻(但不是对角相邻)。 形成一条河流的相邻 1 的数量决定了它的大小。 编写一个 function,它返回一个数组,该数组包含输入矩阵中表示的所有河流的大小。
Input:
array =[
[0, 0, 1, 0, 1, 0],
[1, 0, 0, 1, 1, 0],
[1, 0, 0, 0, 0, 1],
[1, 0, 0, 1, 1, 1],
]
output:
There are 4 rivers in the given matrix, and their sizes are: [1, 3, 3, 4]
但实际上我希望拥有: [1, 2, 2, 2, 3, 3]
因为我只寻找水平或垂直相邻的河流,而不是两者兼而有之的河流。 (正如我的算法 output 带来的)。
我的算法使用 DFS:
def riverSizes(matrix):
sizes = [] # # It would contain all of our river sizes
visited = [[False for value in row] for row in matrix]
for i in range(len(matrix)):
for j in range(len(matrix[i])):
if visited[i][j]:
continue
traverseNode(i, j, matrix, visited, sizes)
return sizes
def traverseNode(i, j, matrix, visited, sizes):
currentRiverSize = 0
nodesToExplore = [[i, j]]
while len(nodesToExplore):
currentNode = nodesToExplore.pop()
i = currentNode[0]
j = currentNode[1]
if visited[i][j]:
continue
visited[i][j] = True
if matrix[i][j] == 0:
continue
currentRiverSize += 1
unvisitedNeighbors = getUnvisitedNeighbors(i, j, matrix, visited)
for neighbor in unvisitedNeighbors:
nodesToExplore.append(neighbor)
if currentRiverSize > 0:
sizes.append(currentRiverSize)
def getUnvisitedNeighbors(i, j, matrix, visited):
unvisitedNeighbors = []
if i > 0 and not visited[i - 1][j]:
unvisitedNeighbors.append([i - 1, j])
if i < len(matrix) - 1 and not visited[i + 1][j]:
unvisitedNeighbors.append([i + 1, j])
if j > 0 and not visited[i][j - 1]:
unvisitedNeighbors.append([i, j - 1])
if j < len(matrix[0]) - 1 and not visited[i][j + 1]:
unvisitedNeighbors.append([i, j + 1])
return unvisitedNeighbors
我该如何解决?
该算法为您提供了问题 [1, 3, 3, 4] 的正确结果。 一条河流由水平或垂直相邻的 1 组成(一个 1 不能同时与另一个 1 水平和垂直相邻,在这种情况下混淆是问题陈述的 'either' 'or' ))。 这就是为什么 getUnvisitedNeighbors 从包含 1 的字段中查看所有垂直和水平方向的原因。换句话说,河流可能有一个 L 形状的结果矩阵
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.