这是我参与8月更文挑战的第28天,活动详情查看:8月更文挑战
难度: 中等
题目描述
给出一个二维整数网格 grid,网格中的每个值表示该位置处的网格块的颜色。
只有当两个网格块的颜色相同,而且在四个方向中任意一个方向上相邻时,它们属于同一连通分量。
连通分量的边界是指连通分量中的所有与不在分量中的正方形相邻(四个方向上)的所有正方形,或者在网格的边界上(第一行/列或最后一行/列)的所有正方形。
给出位于 (r0, c0) 的网格块和颜色 color,使用指定颜色 color 为所给网格块的连通分量的边界进行着色,并返回最终的网格 grid 。
示例 1:
输入:grid = [[1,1],[1,2]], r0 = 0, c0 = 0, color = 3 输出:[[3, 3], [3, 2]]
示例 2:
输入:grid = [[1,2,2],[2,3,2]], r0 = 0, c0 = 1, color = 3 输出:[[1, 3, 3], [2, 3, 3]]
示例 3:
输入:grid = [[1,1,1],[1,1,1],[1,1,1]], r0 = 1, c0 = 1, color = 2 输出:[[2, 2, 2], [2, 1, 2], [2, 2, 2]]
提示:
1 <= grid.length <= 50 1 <= grid[0].length <= 50 1 <= grid[i][j] <= 1000 0 <= r0 < grid.length 0 <= c0 < grid[0].length 1 <= color <= 1000
解题思路
判断四个边是否全,只要有不全就上色
核心思想就是探索当前点的上下左右四个点,如果要探索的点与当前点数字相同,并且没有被探索过,就递归到这个探索点
要用一个数组来记录所有当前探索过的点,因为上下左右同时递归,可能会和之前探索过的点重合,即原来连通区域是一个环的话。因此这步要小心一点。
同时要注意,只染色边界,如果点在边界就直接染色,如果这个点4周都已经染过了,那就不用染了。
class Solution(object):
def colorBorder(self, grid, r0, c0, color):
color_1 = grid[r0][c0]
color_2 = color
a, b = len(grid), len(grid[0])
seen = set()
def dfs(i,j):
if (i,j) in seen: return True
if not (0<=i<a and 0<=j<b and grid[i][j] == color_1):
return False
seen.add((i,j))
if dfs(i+1, j) + dfs(i-1, j) + dfs(i, j+1) + dfs(i, j-1)<4:
grid[i][j] = color_2
return True
dfs(r0, c0)
return grid