Leetcode 994. 腐烂的橘子（BFS模板题）

本文主要是介绍Leetcode 994. 腐烂的橘子（BFS模板题），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Description

在给定的网格中，每个单元格可以有以下三个值之一：值 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 。来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rotting-oranges
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Solution

经典的BFS，代码非常clean，对比以前的C艹模板，虽然思路一样，但不知道简洁多少。

class Solution(object):def orangesRotting(self, grid):R, C = len(grid), len(grid[0])# queue - all starting cells with rotting orangesqueue = collections.deque()for r, row in enumerate(grid): # 用了enumerate小技巧for c, val in enumerate(row):if val == 2:queue.append((r, c, 0))def neighbors(r, c):for nr, nc in ((r-1,c),(r,c-1),(r+1,c),(r,c+1)):if 0 <= nr < R and 0 <= nc < C:yield nr, nc # 生成器是亮点d = 0while queue:r, c, d = queue.popleft()for nr, nc in neighbors(r, c):if grid[nr][nc] == 1:grid[nr][nc] = 2queue.append((nr, nc, d+1))if any(1 in row for row in grid): # any关键字也让代码clean不少return -1return d