在给定的 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

Python实现

宽度优先,这里要使用队列,先记录腐烂的橘子,然后从队列里面取出橘子进行拓展,如果能想到这个,代码就容易写出来了。

class Solution:

    def isValid(self, grid, i,j):
        return 0<=i<len(grid) and 0<=j<len(grid[0])

    def orangesRotting(self, grid: List[List[int]]) -> int:
        m = len(grid)
        n = len(grid[0])

        q = deque()
        for i in range(m):
            for j in range(n):
                if grid[i][j]==2:
                    q.append([i,j,0])
        d=0
        while q:
            row,col, d = q.popleft()
            xy= [[0,1],[0,-1],[1,0],[-1,0]]
            for x,y in xy:
                dx = row+x
                dy = col+y
                if self.isValid(grid,dx,dy):
                    if grid[dx][dy]==1:
                        grid[dx][dy]=2
                        q.append([dx,dy,d+1])
        for i in range(m):
            for j in range(n):
                if grid[i][j]==1:
                    return -1
        return d