问题来源:地图分析
问题描述:你现在手里有一份大小为 N x N 的『地图』(网格) grid,上面的每个『区域』(单元格)都用 0 和 1 标记好了。其中 0 代表海洋,1 代表陆地,你知道距离陆地区域最远的海洋区域是是哪一个吗?请返回该海洋区域到离它最近的陆地区域的距离。
我们这里说的距离是『曼哈顿距离』( Manhattan Distance):(x0, y0) 和 (x1, y1) 这两个区域之间的距离是 |x0 - x1| + |y0 - y1| 。
如果我们的地图上只有陆地或者海洋,请返回 -1。
例子:
输入:
[[1,0,1],
[0,0,0],
[1,0,1]]
输出:2
解释:
海洋区域 (1, 1) 和所有陆地区域之间的距离都达到最大,最大距离为 2。
思路:以所有的陆地为中心,做广度搜索,每前进一步记下步数,这样广度搜索结束步数即为最大距离。参考@lucifer的解答。
class Solution:
def maxDistance(self, grid: List[List[int]]) -> int:
n = len(grid)
steps = -1
# 挑出所有的陆地
queue = [(i, j) for i in range(n) for j in range(n) if grid[i][j] == 1]
# 极端情况
if len(queue) == 0 or len(queue) == n ** 2: return -1
# 广度搜索
while queue:
for _ in range(len(queue)):
x, y = queue.pop(0)
for xi, yi in [(x-1, y), (x+1, y), (x, y-1), (x, y+1)]:
if xi < n and yi < n and xi > -1 and yi > -1 and grid[xi][yi] == 0:
queue.append((xi, yi))
grid[xi][yi] = -1
steps += 1
return steps