【leetcode】994. 腐烂的橘子
程序员文章站
2024-02-21 14:14:57
...
在给定的网格中,每个单元格可以有以下三个值之一:
值 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
这个简单题不像个简单题,也许是我写题生疏了,呜呜。。
中间进行了很多修修补补,提交也测出了很多边界样例。
代码如下:
import copy
class Solution(object):
def orangesRotting(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
is_all_rot = False
times = 0
while not is_all_rot:
grid_back = copy.deepcopy(grid)
is_all_rot = True
is_have_root = False
temp_flag = [[1 for i in range(len(grid[0]))] for j in range(len(grid))]
for i in range(len(grid)):
for j in range(len(grid[0])):
if grid[i][j] == 2:
is_have_root = True
if grid[i][j] == 1:
is_all_rot = False
if (i - 1 > -1 and grid[i-1][j] == 2 and temp_flag[i-1][j]) or (j - 1 > -1 and grid[i][j-1] == 2 and temp_flag[i][j-1]) or \
(i + 1 < len(grid) and grid[i+1][j] == 2 and temp_flag[i+1][j]) or (j + 1 < len(grid[0]) and grid[i][j+1] == 2 and temp_flag[i][j+1]):
grid[i][j] = 2
temp_flag[i][j] = 0
if (i - 1 < 0 or grid[i-1][j] == 0) and (j - 1 < 0 or grid[i][j-1] == 0) and (i + 1 > len(grid) - 1 or grid[i+1][j] == 0) and (j + 1 > len(grid[0]) - 1 or grid[i][j+1] == 0):
return -1
if not is_have_root:
if times == 0 and not is_all_rot:
return -1
break
if grid_back == grid:
if is_have_root and not is_all_rot:
return -1
if not is_all_rot:
times += 1
return times
边界测试样例:
grid = [[2, 1, 1], [1, 1, 0], [0, 1, 1]]
grid = [[0,2]]
grid = [[2,1,1],[0,1,1],[1,0,1]]
grid = [[1],[2],[1],[2]]
grid = [[1], [1], [1], [1]]
grid = [[0]]
grid = [[2],[2],[1],[0],[1],[1]]
今天先这样吧,思路后面再补。
思路可参考官方思路https://leetcode-cn.com/problems/rotting-oranges/solution/