算法刷题111

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;
    }
}