在给定的网格中,每个单元格可以有以下三个值之一:
值 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
class Solution(object):
def orangesRotting(self, grid):
#判断第0分钟就是最后结果的情况
flag = 1
for i in range(len(grid)):
for j in range (len(grid[0])):
if grid[i][j] == 1:
flag = 0
if flag == 1:
return 0
my_queque = []
result = 0
#将初始的坏橘子入队
for i in range(len(grid)):
for j in range (len(grid[0])):
if grid[i][j] == 2:
my_queque.append((i,j))
#对队列中的坏橘子,依次出队,并让其感染相邻的橘子,将新感染的橘子入队,直到队列为空
while my_queque:
count = len(my_queque)
result += 1
for i in range(count):
cur = my_queque.pop(0)
if cur[0]-1 >= 0 and grid[cur[0]-1][cur[1]] == 1:
grid[cur[0]-1][cur[1]] = 2 if grid[cur[0]-1][cur[1]] == 1 else 0
my_queque.append((cur[0]-1,cur[1]))
if cur[0]+1 <len(grid) and grid[cur[0]+1][cur[1]] == 1:
grid[cur[0]+1][cur[1]] = 2 if grid[cur[0]+1][cur[1]] == 1 else 0
my_queque.append((cur[0]+1,cur[1]))
if cur[1] -1 >= 0 and grid[cur[0]][cur[1]-1] == 1:
grid[cur[0]][cur[1]-1] = 2 if grid[cur[0]][cur[1]-1] == 1 else 0
my_queque.append((cur[0],cur[1]-1))
if cur[1]+1 <len(grid[0]) and grid[cur[0]][cur[1]+1] == 1:
grid[cur[0]][cur[1]+1] = 2 if grid[cur[0]][cur[1]+1] == 1 else 0
my_queque.append((cur[0],cur[1]+1))
#如果有未感染的橘子,返回 -1
for i in range(len(grid)):
if 1 in grid[i]:
return -1
#否则,返回 result-1(因为最后一次没有新感染的橘子,但是result+1)
return result-1