leetcode-994 腐烂的橘子 python3 BFS

在给定的网格中，每个单元格可以有以下三个值之一：

值 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