力扣 1034. 边界着色

题目

给你一个大小为 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;
    }
}