2021-12-27每日一题——边界着色

题目

给你一个大小为 m x n 的整数矩阵 grid ，表示一个网格。另给你三个整数 row、col 和 color 。网格中的每个值表示该位置处的网格块的颜色。

当两个网格块的颜色相同，而且在四个方向中任意一个方向上相邻时，它们属于同一 连通分量 。

连通分量的边界 是指连通分量中的所有与不在分量中的网格块相邻（四个方向上）的所有网格块，或者在网格的边界上（第一行/列或最后一行/列）的所有网格块。

请你使用指定颜色 color 为所有包含网格块 grid[row][col] 的 连通分量的边界 进行着色，并返回最终的网格 grid 。

题解

class Solution {
    boolean[][] visited; //记录睡觉哦否访问过当前节点
    int[][] dirs = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}}; //上下左右四个平移方向
    int n, m;
    List<int[]> ans = new ArrayList<>(); //存储结果
    public int[][] colorBorder(int[][] grid, int row, int col, int color) {
        n = grid.length;
        m = grid[0].length;
        visited = new boolean[n][m];
        //从grid[row][col]开始搜索
        int originalColor = grid[row][col];
        visited[row][col] = true;
        dfs(grid, row, col, originalColor);
        for(int i=0;i<ans.size();i++){
            grid[ans.get(i)[0]][ans.get(i)[1]] = color;  //给满足条件的点着色
        }
        return grid;
    }
    //深度优先遍历
    public void dfs(int[][] grid, int x, int y, int originalColor){
        boolean isBorder = false; //记录当前节点是否为边界
        for(int i=0;i<4;i++){
            //平移后的点
            int new_x = dirs[i][0] + x;
            int new_y = dirs[i][1] + y;
            //进行边界判断,当下一个点在界内且颜色不等于开始点的颜色，则当前点不是边界点
            if(!(new_x>=0 && new_x<n && new_y>=0 && new_y<m && grid[new_x][new_y]==originalColor)){
                isBorder = true;
            }
            else if(!visited[new_x][new_y]){
                visited[new_x][new_y] = true;
                dfs(grid, new_x, new_y, originalColor);
            }
        }
        if(isBorder){
            ans.add(new int[]{x, y}); //将满足条件的点放入集合
        }
    }
}