Leetcode 994 Rotting Oranges

发表于 |

在给定的网格中，每个单元格可以有以下三个值之一：
0 代表空单元格；
1 代表新鲜橘子；
2 代表腐烂的橘子。
每过一分钟，任何与腐烂的橘子（在 4 个正方向上）相邻的新鲜橘子都会腐烂。

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

示例 1：
输入：[[2,1,1],[1,1,0],[0,1,1]]
输出：4

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

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

提示：

  1. 1 <= grid.length <= 10
  2. 1 <= grid[0].length <= 10
  3. grid[i][j] 仅为 0、1 或 2

思路分析：
这道题和我博客中bfs详解中一道题（与0的距离）非常相似
首先我们看到题目说了最短，那么应该想到bfs算法找最短
然后起点怎么找呢？因为所有坏橘子会同时影响到其相邻橘子，所以我们将所有的坏橘子作为第1层，然后就是常规bfs，对bfs算法还不太懂或者不够熟练的可以去参考我博客置顶的bfs算法系列详解。

但由于可能存在某个新鲜橘子始终无法被腐烂，所以这里对初始的新鲜橘子进行计数，每腐烂一个就减一，观察其数量最后是否为0。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution(object):
    def orangesRotting(self, grid):
        n,m = len(grid), len(grid[0])
        Q = collections.deque([])
        cnt = 0
        for i in range(n):
            for j in range(m):
                if grid[i][j] == 1: cnt += 1
                if grid[i][j] == 2: Q.append((i,j))
        res = 0
        while Q:
            for _ in range(len(Q)):
                i,j = Q.popleft()
                for x, y in [(i+1,j), (i-1,j), (i,j+1), (i,j-1)]:
                    if 0<=x<n and 0<=y<m and grid[x][y] == 1:
                        grid[x][y] = 2
                        cnt -= 1
                        Q.append((x,y))
            res += 1
        return max(0, res-1) if cnt == 0 else -1