1 题目描述
2 解题思路
2.1 多源广搜
使用广搜的思想,但是和一般的广搜不同,这里一开始的队列不是一个元素,而是一开始就腐烂的所有橘子的坐标。
同时我们队列里面的每一个元素是一个三元组,分别是腐烂橘子的坐标,和当前已经经过了的分钟数。对于每个已经腐烂的橘子,我们向外找一圈,将新鲜的橘子变成腐烂的橘子,然后原本新鲜位置橘子的坐标入队列。
当队列为空,表示所有橘子都已经考虑过了。这时候我们看是否还有新鲜的橘子,有的话返回-1
class Solution:
def orangesRotting(self, grid: List[List[int]]) -> int:
lst=[]
num_1=0
row=len(grid)
column=len(grid[0])
for i in range(row):
for j in range(column):
if(grid[i][j]==1):
num_1+=1
elif(grid[i][j]==2):
lst.append((i,j,0))
time=0
while(lst):
i,j,time=lst.pop(0)
for (x,y) in [(i-1,j),(i+1,j),(i,j-1),(i,j+1)]:
if(x>=0 and x<row and y>=0 and y<column):
if(grid[x][y]==1):
grid[x][y]=2
num_1-=1
lst.append((x,y,time+1))
if(num_1!=0):
return(-1)
else:
return(time)
2.2 while 循环
我们记当前新鲜橘子的数量为num_1,如果这个数字不为0,且连续两分钟内这个数字在变化,说明腐烂仍在继续,那么就继续循环。
每次我们遍历当前腐烂的橘子,让它周围一圈的新鲜橘子也腐烂。如果一个腐烂的橘子已经让周围一圈新鲜的橘子腐烂了,那我们将他的状态设置为-1,这样下一轮遍历已经腐烂的橘子的时候,这些橘子就不会再考虑了
class Solution:
def orangesRotting(self, grid: List[List[int]]) -> int:
num_1=0
for i in grid:
num_1+=i.count(1)
row=len(grid)
column=len(grid[0])
num=0
while(num_1!=0):
prev_num=num_1
num+=1
tmp_lst=[]
for i in range(row):
for j in range(column):
if(grid[i][j]==2):
grid[i][j]=-1
tmp_lst.append((i,j))
#tmp_lst中是当前这一轮腐烂的橘子,这些橘子的状态被设置为-1,之后将不再考虑之
for (i,j) in tmp_lst:
for (x,y) in [(i-1,j),(i+1,j),(i,j-1),(i,j+1)]:
if(x>=0 and x<row and y>=0 and y<column):
if(grid[x][y]==1):
grid[x][y]=2
num_1-=1
if(prev_num==num_1):
return -1
#如果连续两分钟,新鲜的橘子数量一样,而且数量不为0,返回-1
return num