给出一个二维整数网格 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
思路:
BFS往外找,找到符合要求的就变色,用visited记录当前点有没有来过,用record记录原始矩阵(因为过程中会修改原始矩阵)。
class Solution(object):
def colorBorder(self, grid, r0, c0, color):
"""
:type grid: List[List[int]]
:type r0: int
:type c0: int
:type color: int
:rtype: List[List[int]]
"""
m= len(grid)
if not m:
return grid
n = len(grid[0])
if not n:
return grid
dx = [1, -1, 0, 0]
dy = [0, 0, -1, 1]
originalcolor = grid[r0][c0]
record = list()
visited = list()
for row in grid:
record.append(row[:])
visited.append(row[:])
def checkNeibor(x, y):
#返回4说明不染色,返回其他说明满足染色条件
if x in [0, m - 1] or y in [0, n - 1]:#在边界上
return -1
cnt = 0
for j in range(4):
xx = x + dx[j]
yy = y + dy[j]
if 0 <= xx < m and 0<= yy < n and originalcolor == record[xx][yy]:
cnt += 1
return cnt
if checkNeibor(r0, c0) != 4:
grid[r0][c0] = color
queue = [[r0,c0]]
while(queue):
newqueue = list()
for node in queue:
x0 = node[0]
y0 = node[1]
for k in range(4):
x = x0 + dx[k]
y = y0 + dy[k]
if 0 <= x < m and 0<= y < n and originalcolor == record[x][y] and visited[x][y] != -1:
newqueue.append([x, y])
visited[x][y] = -1
# print [x, y], queue
if checkNeibor(x, y) != 4:
grid[x][y] = color
queue = newqueue[:]
return grid