给你一个大小为 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
java
class Solution {
public int[][] colorBorder(int[][] grid, int row, int col, int color) {
int m = grid.length;
int n = grid[0].length;
ArrayList<Pair<Integer, Integer>> arr = new ArrayList<>();
Queue<Pair<Integer, Integer>> que = new LinkedList<>();
que.add(new Pair<>(row, col));
arr.add(new Pair<>(row, col));
int[][] flag = new int[m][n];
flag[row][col] = 1;
int dy[] = {1, -1, 0, 0};
int dx[] = {0, 0, 1, -1};
while (!que.isEmpty()) {
Pair<Integer, Integer> p = que.poll();
int r = p.getKey();
int c = p.getValue();
for (int k = 0; k < 4; k++) {
int y = r + dy[k];
int x = c + dx[k];
if (y >= 0 && y < m && x >= 0 && x < n && 0 == flag[y][x] && grid[y][x] == grid[r][c]) {
flag[y][x] = 1;
que.add(new Pair<>(y, x));
arr.add(new Pair<>(y, x));
}
}
}
int[][] tmp = new int[m][n];
for(int i=0;i<m;i++) {
for(int j=0;j<n;j++) {
tmp[i][j]=grid[i][j];
}
}
for (Pair<Integer, Integer> p : arr) {
int r = p.getKey();
int c = p.getValue();
if (r == 0 || r == m - 1 || c == 0 || c == n - 1) {
grid[r][c] = color;
} else {
for (int k = 0; k < 4; k++) {
int y = r + dy[k];
int x = c + dx[k];
if (tmp[y][x] != tmp[r][c]) {
grid[r][c] = color;
}
}
}
}
return grid;
}
}