994. Rotting Oranges [Easy] 宽度优先

最新推荐文章于 2022-03-21 21:49:46 发布

原创最新推荐文章于 2022-03-21 21:49:46 发布 · 159 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种解决“腐烂橘子”问题的算法，通过记录新鲜橘子数量并利用队列跟踪腐烂过程，实现橘子腐烂状态的扩散更新。详细解析了算法流程与Python代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

994. Rotting Oranges

首先，记录下所有1的个数，并把所有2的放入一个队列，记录步数，每一步把队列里面相邻的1换成2，更新1的个数，并把腐烂的放入队列。

代码如下：

class Solution(object):
    def orangesRotting(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        fresh = 0
        qu = []
        m, n = len(grid), len(grid[0])
        for i in range(m):
            for j in range(n):
                if grid[i][j] == 1:
                    fresh += 1
                elif grid[i][j] == 2:
                    qu.append((i,j))
        if fresh == 0:
            return 0
        step = 0
        direction = [(0,1), (0,-1), (1,0), (-1,0)]
        while qu:
            step += 1
            for i in range(len(qu)):
                temp = qu.pop(0)
                for j in range(4):
                    x = temp[0] + direction[j][0]
                    y = temp[1] + direction[j][1]
                    if x>=0 and y>=0 and x<m and y<n and grid[x][y]==1:
                        fresh -= 1
                        grid[x][y] = 2
                        qu.append((x,y))
                        if fresh == 0:
                            return step
        return -1