算法的重要性，我就不多说了吧，想去大厂，就必须要经过基础知识和业务逻辑面试+算法面试。所以，为了提高大家的算法能力，后续每天带大家做一道算法题，题目就从LeetCode上面选 ！今天和大家聊的问题叫做 墙和门，我们先来看题面：​LeetCode刷题实战286：墙和门_刷题你被给定一个 m × n 的二维网格，网格中有以下三种可能的初始化值：

  • -1 表示墙或是障碍物

  • 0 表示一扇门

  • INF 无限表示一个空的房间。然后，我们用 231 - 1 = 2147483647 代表 INF。你可以认为通往门的距离总是小于 2147483647 的。

你要给每个空房间位上填上该房间到 最近 门的距离，如果无法到达门，则填 INF 即可。

示例

 

给定二维网格：
INF -1  0 INF
INF INF INF -1
INF -1 INF -1
0  -1 INF INF
运行完你的函数后，该网格应该变成：
3 -1 0 1
2 2 1 -1
1 -1 2 -1
0 -1 3 4

 

解题

从门开始宽度优先搜索，环一层一层扩散，能被门的光环覆盖到的时候，就是距离门的最短距离。这非常直接。前面写题快有一种思维定式是，只有DFS才有上下左右四个探索，其实BFS也一样可以有递归的写法。要灵活掌握算法背后的思想。本题用BFS的解法是最简单易懂的。

 

class Solution:
    def wallsAndGates(self, rooms: List[List[int]]) -> None:
        """
        Do not return anything, modify rooms in-place instead.
        """
        if not rooms or len(rooms) == 0:
            return 
        row = len(rooms)
        col = len(rooms[0])
        
        def bfs(rooms, i, j, val): # val表示环扩展的距离
            nonlocal row, col
            if i < 0 or i >= row or j < 0 or j >= col:
                return
            if rooms[i][j] < val: # 如果当前位置已经找到更近的门了，直接返回
                return
            rooms[i][j] = val
            
            bfs(rooms, i + 1, j, val + 1)
            bfs(rooms, i, j + 1, val + 1)
            bfs(rooms, i - 1, j, val + 1)
            bfs(rooms, i, j - 1, val + 1)
        for i in range(row):
            for j in range(col):
                if rooms[i][j] == 0:
                    bfs(rooms, i, j, 0)

好了，今天的文章就到这里，如果觉得有所收获，你们的支持是我最大的动力 。

 

​LeetCode刷题实战286：墙和门_​LeetCode_02