​LeetCode刷题实战286：墙和门

算法的重要性，我就不多说了吧，想去大厂，就必须要经过基础知识和业务逻辑面试+算法面试。所以，为了提高大家的算法能力，这个公众号后续每天带大家做一道算法题，题目就从LeetCode上面选 ！

今天和大家聊的问题叫做 墙和门，我们先来看题面：

https://leetcode-cn.com/problems/walls-and-gates/

你被给定一个 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

解题

https://blog.csdn.net/u011240016/article/details/100365250

从门开始宽度优先搜索，环一层一层扩散，能被门的光环覆盖到的时候，就是距离门的最短距离。

这非常直接。

前面写题快有一种思维定式是，只有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)

好了，今天的文章就到这里，如果觉得有所收获，请顺手点个在看或者转发吧，你们的支持是我最大的动力 。

上期推文：

LeetCode1-280题汇总，希望对你有点帮助！

LeetCode刷题实战281：锯齿迭代器

LeetCode刷题实战282：给表达式添加运算符

LeetCode刷题实战283：移动零

LeetCode刷题实战284：顶端迭代器

LeetCode刷题实战285：二叉搜索树中的顺序后继