Python Leetcode(994.腐烂的橘子)

Python Leetcode(994.腐烂的橘子)

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

值 0 代表空单元格；
值 1 代表新鲜橘子；
值 2 代表腐烂的橘子。

每分钟，任何与腐烂的橘子（在 4 个正方向上）相邻的新鲜橘子都会腐烂。

返回直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能，返回 -1。

Solution:(由于坏橘子感染过之后就不用再考虑它了，可以使用队列，先把坏橘子的坐标添加到队列当中，每次取出队列的第一个坏橘子，对它周围的橘子进行判断，若为好橘子则感染，将其坐标添加进队列当中[注意这里要把橘子对应的时间也添加进去，防止重复记录时间]，一直循环到队列为空为止。最后遍历数组，若有好橘子则返回-1，否则返回time)

class Solution(object):
    def orangesRotting(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        x, y, time = len(grid), len(grid[0]), 0
        queue = []
        # 定义一个移动坐标的数组
        locs = [[-1, 0], [1, 0], [0, 1], [0, -1]]
        
        for i in range(x):
            for j in range(y):
                if grid[i][j] == 2:
                    queue.append((i, j, time))
        
        while queue:
            loc_x, loc_y, time = queue.pop(0)
            for loc in locs:
                sur_x = loc_x + loc[0]
                sur_y = loc_y + loc[1]
                if 0 <= sur_x < x and 0 <= sur_y < y and grid[sur_x][sur_y] == 1:
                    grid[sur_x][sur_y] = 2
                    queue.append((sur_x, sur_y, time+1))
        
        for row in grid:
            if 1 in row:
                return -1
        return time

solution = Solution()
print(solution.orangesRotting([[2,1,1],[1,1,0],[0,1,1]]))

4