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