考點:隊列
在給定的網(wǎng)格中榆浓,每個單元格可以有以下三個值之一:
值 0 代表空單元格袖迎;
值 1 代表新鮮橘子闪盔;
值 2 代表腐爛的橘子潜秋。
每分鐘,任何與腐爛的橘子(在 4 個正方向上)相鄰的新鮮橘子都會腐爛。
返回直到單元格中沒有新鮮橘子為止所必須經(jīng)過的最小分鐘數(shù)茴晋。如果不可能,返回 -1回窘。
動態(tài)的遍歷queue
廣度優(yōu)先搜索算法
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