leetcode994.腐烂的橘子「2020-0304」
程序员文章站
2022-05-21 12:09:11
...
1.题目描述
在给定的网格中,每个单元格可以有以下三个值之一:
值 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 。
提示:
1 <= grid.length <= 10
1 <= grid[0].length <= 10
grid[i][j] 仅为 0、1 或 2来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotting-oranges
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.解题思路
bfs,模板题,注意对被腐烂的橘子更新状态即可
3.代码实现
class Solution(object):
def orangesRotting(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
m = len(grid)
n = len(grid[0])
dirs = ((1,0),(-1,0),(0,1),(0,-1))
Q = []
visited = set()
minute = 0
for i in range(m):
for j in range(n):
if grid[i][j] == 2:
Q.append([i,j,minute])
visited.add((i,j))
while Q:
i,j,minute = Q.pop(0)
for k in range(4):
x = i+dirs[k][0]
y = j+dirs[k][1]
if 0 <= x < m and 0 <= y < n and (x,y) not in visited and grid[x][y]!=0:
grid[x][y] = 2
Q.append((x,y,minute+1))
visited.add((x,y))
for i in range(m):
for j in range(n):
if grid[i][j] == 1:
return -1
return minute