题目
给你一个大小为 m x n 的整数矩阵 grid ,表示一个网格。另给你三个整数 row、col 和 color 。网格中的每个值表示该位置处的网格块的颜色。
当两个网格块的颜色相同,而且在四个方向中任意一个方向上相邻时,它们属于同一 连通分量 。
连通分量的边界 是指连通分量中的所有与不在分量中的网格块相邻(四个方向上)的所有网格块,或者在网格的边界上(第一行/列或最后一行/列)的所有网格块。
请你使用指定颜色 color 为所有包含网格块 grid[row][col] 的 连通分量的边界 进行着色,并返回最终的网格 grid 。
示例
输入:grid = [[1,1],[1,2]], row = 0, col = 0, color = 3
输出:[[3,3],[3,2]]
输入:grid = [[1,2,2],[2,3,2]], row = 0, col = 1, color = 3
输出:[[1,3,3],[2,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]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/coloring-a-border
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:BFS
1、不直接在 grid数组 上面更改,初始化一个ans数组,用来存储修改的结果。
2、从 grid[row][col]开始遍历。如果 grid[row][col]的上下左右4个方向都和它的颜色一致,则grid[row][col]不是边界。否则上色。
Java实现
class Solution {
int[] directions = new int[]{-1, 0, 1, 0, -1};
public int[][] colorBorder(int[][] grid, int row, int col, int color) {
Queue<int[]> q = new LinkedList<>();
q.offer(new int[]{row, col});
int m = grid.length, n = grid[0].length;
int[][] ans = new int[m][n];
while (!q.isEmpty()) {
int sz = q.size();
for (int i = 0; i < sz; i++) {
int[] cur = q.poll();
int cnt = 0;
for (int j = 0; j < directions.length - 1; j++) {
int x = cur[0] + directions[j];
int y = cur[1] + directions[j + 1];
//超出矩阵范围,出界
if (x >= m || x < 0 || y >= n || y < 0) continue;
//如果上下左右和中间的颜色一样,cnt = 4,则不是边界
if (grid[cur[0]][cur[1]] != grid[x][y]) continue;
else cnt++;
//已经被填充过颜色
if(ans[x][y] != 0) continue;
q.offer(new int[]{x, y});
}
//不是边界就填充原本的颜色,否则上色。
ans[cur[0]][cur[1]] = cnt == 4 ? grid[cur[0]][cur[1]] : 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;
}
}