2021.12.7 力扣-每日一题-边界着色

题目描述：

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

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

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

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

样例：

方法一：

class Solution {
public:
    int dire[4][2] = {{1, 0}, {0, -1}, {-1, 0}, {0, 1}};
    vector<vector<int>> colorBorder(vector<vector<int>>& grid, int row, int col, int color) {
        int maincolor = grid[row][col]; //所要找的连通分量的颜色
        queue<pair<int, int>> que;      //队列，存储颜色是maincolor的网格块坐标
        int m = grid.size();
        int n = grid[0].size();
        vector<vector<int>> ans(m, vector<int>(n, 0));
        ans[row][col] = color;
        que.emplace(row, col);
        
        while (!que.empty())
        {
            int x = que.front().first;
            int y = que.front().second;
            que.pop();
            int cnt = 0;                //记录（x，y）这个网格块的上下左右有几个相同颜色的块
            for (int i = 0; i < 4; i++)
            {
                int curx = x + dire[i][0];
                int cury = y + dire[i][1];
                if (curx >= 0 && curx < m && cury >= 0 && cury < n && grid[curx][cury] == maincolor)
                {
                    cnt++;
                    //如果该点还未被访问过，就加入队列
                    if (ans[curx][cury] == 0)
                    {
                        que.emplace(curx, cury);
                    }
                }
            }
            //当（x，y）的上下左右都是相同颜色的块时，说明（x，y）不是连通分量的边界
            //cnt != 4有两种情况，一种是（x，y）的上下左右至少有一个不是相同颜色；另一种是（x，y）是网格的边界，也必然是连通分量的边界
            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;
    }
};

还以为是并查集的方法，毕竟跟见过的岛屿问题有点像，最后看题解原来是用bfs或者dfs。。

自己写时用方向数组倒是写对了，但是却不知道怎么判断一个网格块是边界，其实很简单，只要这个网格块属于要找的连通分量并且上下左右四个网格块中，如果4个块都是相同颜色的块，那么该网格块就不是边界，反之即是边界（看代码中的注释）。

每次都是看完题解才发现居然这么简单，这类题目还是得画画图的。