在給定的網格中,每個單元格可以有以下三個值之一:
- 值
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
思路:分別保存好的和壞的座標,然後做遞歸。
class Solution(object):
def orangesRotting(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
len1 = len(grid)
if len1 == 0:
return 0
len2 = len(grid[0])
if len2 == 0:
return 0
self.num = 0
self.grid = grid
bad = []
self.good = []
for i in range(len1):
for j in range(len2):
if grid[i][j] == 2:
bad.append([i,j])
if grid[i][j] == 1:
self.good.append([i,j])
if not self.good:
return 0
if not bad:
return -1
def dfs(bad):
badx = []
for i in bad:
x = i[0]
y = i[1]
if x > 0:
if self.grid[x-1][y] == 1:
self.grid[x-1][y] = 2
badx.append([x-1, y])
self.good.remove([x-1, y])
if x < len1-1:
if self.grid[x+1][y] == 1:
self.grid[x+1][y] = 2
badx.append([x+1, y])
self.good.remove([x+1, y])
if y > 0:
if self.grid[x][y-1] == 1:
self.grid[x][y-1] = 2
badx.append([x, y-1])
self.good.remove([x, y-1])
if y < len2-1:
if self.grid[x][y+1] == 1:
self.grid[x][y+1] = 2
badx.append([x, y+1])
self.good.remove([x, y+1])
if badx == []:
return
self.num += 1
dfs(badx)
dfs(bad)
if self.good:
return -1
return self.num