每日一道leetcode（python）994. 腐烂的橘子

每日一道leetcode（python）994. 腐烂的橘子

2021-08-25

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

值 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 <= grid.length <= 10
1 <= grid[0].length <= 10
grid[i][j] 仅为 0、1 或 2

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rotting-oranges
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

将所有的烂橘子的坐标存放到一个队列里面，然后通过BFS将附近的新鲜橘子腐蚀即可

class Solution:
    def orangesRotting(self, grid: List[List[int]]) -> int:
        r, c = len(grid), len(grid[0])
        d = [0, 1, 0, -1, 0]
        q = [] # 用列表来表示队列
        step = cnt = 0
        for i in range(r):
            for j in range(c):
                if grid[i][j] == 2:
                    q.append([i, j])
                if grid[i][j] == 1:
                    cnt += 1
        while q: # 只要队列不空就循环
            if cnt == 0:
                return step
            n = len(q)
            for i in range(n):
                x, y  = q.pop(0) # 出队
                for j in range(4):
                    nx, ny = x + d[j], y + d[j + 1]
                    if 0 <= nx < r and 0 <= ny < c and grid[nx][ny] == 1:
                        cnt -= 1
                        q.append([nx, ny])
                        grid[nx][ny] = 2
            step += 1
        return step if cnt == 0 else -1