题目
在给定的网格中,每个单元格可以有以下三个值之一:
- 值
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
-
方案一(BFS)
class Solution:
def orangesRotting(self, grid: List[List[int]]) -> int:
if not grid or not grid[0]:
return 0
self.m = len(grid)
self.n = len(grid[0])
# BFS
starts = []
n = 0 # 未腐烂橘子的个数
for i in range(self.m):
for j in range(self.n):
if grid[i][j] == 2:
starts.append((i, j))
elif grid[i][j] == 1:
n += 1
if not starts:
if n == 0: # 没有腐烂的橘子,且没有新鲜的橘子
return 0
return -1
deque = collections.deque(starts)
minutes = 0
while deque:
length = len(deque)
minutes += 1
for _ in range(length):
x, y = deque.popleft() # 已腐烂的橘子的坐标
for next_x, next_y in self.surround(x, y):
if grid[next_x][next_y] == 1:
grid[next_x][next_y] = 2
deque.append((next_x, next_y))
for i in range(self.m):
for j in range(self.n):
if grid[i][j] == 1:
return -1
return minutes - 1
def surround(self, x, y):
if x > 0:
yield x - 1, y
if x < self.m - 1:
yield x + 1, y
if y > 0:
yield x, y - 1
if y < self.n - 1:
yield x, y + 1
原文
https://leetcode-cn.com/explore/interview/card/2020-top-interview-questions/288/graph/1291/