994. 腐烂的橘子
在给定的网格中,每个单元格可以有以下三个值之一:
值 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
思路就是BFS,但是限制条件需要特别注意一下:
- 初始的时候,全部都是新鲜的,返回 -1
- 初始的时候,全部都是坏, 或者没有新鲜的,也没有坏的,返回 0
- 最后BFS之后还有新鲜的,返回-1
- 否则,返回计算的过程数,注意要减一,因为最后一轮坏的需要在BFS弹出一遍。
class Solution:
def orangesRotting(self, grid: List[List[int]]) -> int:
m, n = len(grid), len(grid[0])
queue = collections.deque()
Fresh = 0
for i in range(m):#将腐烂的加入队列,并且看是否还有1
for j in range(n):
if grid[i][j] == 2:
queue.append((i,j))
if grid[i][j] == 1:
Fresh = 1#存在新鲜的
#全是新鲜的 -1, 没有新鲜的也没有坏的0, 只有坏的 0
if not Fresh: return 0
elif Fresh and not queue: return -1
cnt = 0
while queue:
cnt += 1
for _ in range(len(queue)):
i, j = queue.popleft()
for x,y in [(i,j+1), (i,j-1), (i-1,j), (i+1,j)]:
if 0 <= x < m and 0 <= y < n and grid[x][y] == 1:
# print(x,y,cnt)
grid[x][y] = 2
queue.append((x,y))
#最后看下是否还有新鲜的
for i in range(m):
for j in range(n):
if grid[i][j] == 1:return -1
return cnt-1