queue:994.腐烂的橘子

考点:队列

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

值 0 代表空单元格;
值 1 代表新鲜橘子;
值 2 代表腐烂的橘子。
每分钟,任何与腐烂的橘子(在 4 个正方向上)相邻的新鲜橘子都会腐烂。

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

动态的遍历queue

  • 遍历queue的同时会追加元素

广度优先搜索算法

class Solution(object):
    def orangesRotting(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        
        d=0
        y_max = len(grid)
        x_max = len(grid[0])
        queue = collections.deque()

        find_flag = False
        for indexy, y in enumerate(grid):
            for indexx, x in enumerate(y):
                if x == 2:
                    queue.append((indexx, indexy, 0))

        def is_in_grid(x, y):
            for x,y in [(x-1,y), (x+1,y), (x,y+1), (x,y-1)]:
                if 0 <= x < x_max and 0 <= y < y_max:
                    yield x,y
        
        while queue:
            x, y, d = queue.popleft()
            
            for xx, yy in is_in_grid(x, y):
                if grid[yy][xx] == 1:
                    grid[yy][xx] = 2
                    queue.append((xx, yy, d+1))

        if any(1 in row for row in grid):
            return -1
        return d

你可能感兴趣的:(queue:994.腐烂的橘子)