刷题记录 HOT100 图论-2：994. 腐烂的橘子

题目：994. 腐烂的橘子

难度：简单

在给定的 m x n 网格 grid 中，每个单元格可以有以下三个值之一：

• 值 0 代表空单元格；
• 值 1 代表新鲜橘子；
• 值 2 代表腐烂的橘子。

每分钟，腐烂的橘子 周围 4 个方向上相邻 的新鲜橘子都会腐烂。

返回 直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能，返回 -1 。

示例 1：

输入：grid = [[2,1,1],[1,1,0],[0,1,1]]
输出：4

示例 2：

输入：grid = [[2,1,1],[0,1,1],[1,0,1]]
输出：-1
解释：左下角的橘子（第 2 行， 第 0 列）永远不会腐烂，因为腐烂只会发生在 4 个方向上。

示例 3：

输入：grid = [[0,2]]
输出：0
解释：因为 0 分钟时已经没有新鲜橘子了，所以答案就是 0 。

提示：

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 10
• grid[i][j] 仅为 0、1 或 2

一、模式识别

1.图论

周围相邻 》 BFS

每分钟传染一次 》 层序

二、代码实现

1.传染结束后筛查未感染橘子

class Solution:
    def orangesRotting(self, grid: List[List[int]]) -> int:
        dirs = [(0, 1), (1, 0), (0, -1), (-1, 0)]
        que, rest = collections.deque(), set()
        m, n = len(grid), len(grid[0])
        for i in range(m):
            for j in range(n):
                if grid[i][j] == 2: que.append((i, j))

        ans = 0
        while que:
            length = len(que)
            for _ in range(length):
                i, j = que.popleft()
                for di, dj in dirs:
                    ti, tj = i + di, j + dj
                    if (0 <= ti < m and 0 <= tj < n) and grid[ti][tj] == 1: 
                        grid[ti][tj] = 2
                        que.append((ti, tj))
            if que: ans += 1

        return ans if not any(1 in row for row in grid) else -1

2.将所有未感染句子放入集合

class Solution:
    def orangesRotting(self, grid: List[List[int]]) -> int:
        dirs = [(0, 1), (1, 0), (0, -1), (-1, 0)]
        que, rest = collections.deque(), set()
        m, n = len(grid), len(grid[0])
        for i in range(m):
            for j in range(n):
                if grid[i][j] == 2: que.append((i, j))
                if grid[i][j] == 1: rest.add((i, j))
        if not rest: return 0
        ans = -1
        while que:
            ans += 1
            length = len(que)
            for _ in range(length):
                i, j = que.popleft()
                for di, dj in dirs:
                    ti, tj = i + di, j + dj
                    if (ti, tj) in rest: 
                        que.append((ti, tj))
                        rest.remove((ti, tj))
        return ans if not rest else -1