leecode的算法刷题
12.7日
1.leetcode每日一题 1034. 边界着 bfs与两个小技巧
leetcode每日一题 1034. 边界着 bfs与两个小技巧
/*
构造 ansans 矩阵作为答案,同时 ansans 也作为判重数组使用(通过判断 ans[i][j]ans[i][j] 是否为 00 来得知是否被处理);
起始时,将 (row, col)(row,col) 位置进行入队,每次从队列中取出元素进行「四联通拓展」:
拓展格子必须与起点格子处于同一「连通分量」,即满足两者起始颜色相同;
进行「四联通拓展」的同时,记录当前出队是否为边界格子。若为边界格子,则使用 colorcolor 进行上色;
跑完 BFS 后,对 ansans 进行遍历,将未上色(ans[i][j] = 0ans[i][j]=0)的位置使用原始色(grid[i][j]grid[i][j])进行上色。
*/
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;
if (grid[x][y] != grid[nx][ny]) continue;
else cnt++;
if (ans[nx][ny] != 0) continue;
d.addLast(new int[]{nx, ny});
}
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;
}
}