leetcode刷题记录11-1034. 边界着色

问题描述

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

如果两个方块在任意 4 个方向上相邻，则称它们 相邻 。

如果两个方块具有相同的颜色且相邻，它们则属于同一个 连通分量 。

连通分量的边界 是指连通分量中满足下述条件之一的所有网格块：

• 在上、下、左、右任意一个方向上与不属于同一连通分量的网格块相邻
• 在网格的边界上（第一行/列或最后一行/列）

请你使用指定颜色 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

问题分析：

典型的dfs模板问题（当然也可以用bfs但我太菜了只有dfs比较熟练一点所以用的dfs），这道题主要是要考虑分两步来算，首先第一步先用dfs，标记所有的连通分量，然后第二步判断是不是连通分量的边界。

代码如下：

class Solution {
public:
    //四个方向
    int dir[4][2] = {0, 1, 0, -1, 1, 0, -1, 0};
    vector<vector<int>> colorBorder(vector<vector<int>>& grid, int r0, int c0, int color) {
        int initialColor = grid[r0][c0];
        //创建一个二维bool数组来标记连通分量
        vector<vector<bool>> visited(grid.size(), vector<bool>(grid[0].size(), false));
        dfs(grid, visited, r0, c0, initialColor);

        //拷贝grid->res，这步比较关键防止修改过程导致isBorder函数的返回值产生变化导致结果错误
        vector<vector<int>> res = grid;
        for (int i = 0; i < grid.size(); i++) {
            for (int j = 0; j < grid[0].size(); j++) {
                //如果是连通分量并且属于边界情况
                if (visited[i][j] && isBorder(grid, i, j, initialColor)) {
                    res[i][j] = color;
                }
                else
                    res[i][j] = grid[i][j];
            }
        }

        return res;
    }

private:
    void dfs(vector<vector<int>>& grid, vector<vector<bool>>& visited, int r, int c, int color) {
        //越界或者不和起始点同一个颜色或者访问过
        if (r < 0 || r >= grid.size() || c < 0 || c >= grid[0].size() || grid[r][c] != color || visited[r][c]) {
            return;
        }
        //标记连通分量
        visited[r][c] = true;
        //dfs遍历四个方向
        dfs(grid, visited, r + 1, c, color);
        dfs(grid, visited, r - 1, c, color);
        dfs(grid, visited, r, c + 1, color);
        dfs(grid, visited, r, c - 1, color);
    }

    bool isBorder(vector<vector<int>>& grid, int r, int c, int color) {
        //边界情况
        if (r == 0 || r == grid.size() - 1 || c == 0 || c == grid[0].size() - 1) {
            return true; 
        }
        //和初始点颜色不一样，等价于不属于同一个连通分量
        if (grid[r-1][c] != color || grid[r+1][c] != color || grid[r][c-1] != color || grid[r][c+1] != color){
            return true; 
        }

        return false;
    }
};