题目描述:
给你一个大小为 m x n 的整数矩阵 grid ,表示一个网格。另给你三个整数 row、col 和 color 。网格中的每个值表示该位置处的网格块的颜色。
当两个网格块的颜色相同,而且在四个方向中任意一个方向上相邻时,它们属于同一 连通分量 。
连通分量的边界 是指连通分量中的所有与不在分量中的网格块相邻(四个方向上)的所有网格块,或者在网格的边界上(第一行/列或最后一行/列)的所有网格块。
请你使用指定颜色 color 为所有包含网格块 grid[row][col] 的 连通分量的边界 进行着色,并返回最终的网格 grid 。
示例 1:
输入:grid = [[1,1],[1,2]], row = 0, col = 0, color = 3
输出:[[3,3],[3,2]]
示例 2:
输入:grid = [[1,2,2],[2,3,2]], row = 0, col = 1, color = 3
输出:[[1,3,3],[2,3,3]]
示例 3:
输入:grid = [[1,1,1],[1,1,1],[1,1,1]], row = 1, col = 1, color = 2
输出:[[2,2,2],[2,1,2],[2,2,2]]
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 50
1 <= grid[i][j], color <= 1000
0 <= row < m
0 <= col < n
思路:
题目的意思就是,给我们一个起始点,根据这个起始点来寻找连通边量的分界。
连通分量的意思就是这个点的值和它上下左右的值都是和所给定的起始位置一样的值。
如果上下左右有一个不是一样的值,或者到达了边界,那就把其涂成给定的color。
如下图所示。红色的区域全都要涂成给定的color值。
广搜和深搜都行。
从给定的起始点开始广搜,然后假如到达的点和起始值的点的值一样,加入广搜队列中。
如果到达的点的上下左右的值都是和起始点的值一样,那么该值不变。
如果到达的点的上下左右的值有一个不一样,或者到达了边界,该值变为给定的color值。
class Solution {
public int[][] colorBorder(int[][] grid, int row, int col, int color) {
int m = grid.length, n = grid[0].length;
int[][] ans = new int[m][n];//结果数组
int[][] dirs = new int[][]{{1,0}, {-1,0}, {0,1}, {0,-1}};//移动数组
Deque<int[]> d = new ArrayDeque<>();//广搜队列
d.addLast(new int[]{row, col});//把起始点加入队列中,开始广搜
while (!d.isEmpty()) {
int[] poll = d.pollFirst();
int x = poll[0], y = poll[1], cnt = 0;
for (int[] di : dirs) {//从上下左右四个方向走
int nx = x + di[0], ny = y + di[1];
if (nx < 0 || nx >= m || ny < 0 || ny >= n) continue;//如果到达边界 continue
if (grid[x][y] != grid[nx][ny]) continue;//如果该值不等于初始值 continue
else cnt++;
if (ans[nx][ny] != 0) continue;//如果该值已经到达过了 continue
d.addLast(new int[]{nx, ny});//如果以上条件都不成立,该点加入广搜队列中
}
//如果cnt==4也就是说上下左右都是初始值,则该点为连通分量,值保持不变。
//如果cnt!=4也就是说该点为连通边界,该点变为指定的color值
ans[x][y] = cnt == 4 ? grid[x][y] : color;
}
//将其他没到达的值从给定的矩阵中复制过来,并返回结果数组
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (ans[i][j] == 0) ans[i][j] = grid[i][j];
}
}
return ans;
}
}