题目描述:
给你一个大小为 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个块都是相同颜色的块,那么该网格块就不是边界,反之即是边界(看代码中的注释)。
每次都是看完题解才发现居然这么简单,这类题目还是得画画图的。
|