994. 腐爛的橘子 難度 簡單

在給定的網格中,每個單元格可以有以下三個值之一:

  • 值 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. 1 <= grid.length <= 10
  2. 1 <= grid[0].length <= 10
  3. grid[i][j] 僅爲 01 或 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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章