leetcode 1034. 边界着色

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