文巾解题 994. 腐烂的橘子

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