题目描述
给你一个大小为 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/coloring-a-border
方法一:
深度优先搜索dfs
class Solution {
public:
vector<vector<int>> colorBorder(vector<vector<int>>& grid, int row, int col, int color) {
int m=grid.size(),n=grid[0].size(),s=grid[row][col];
vector<vector<bool>> visit(m, vector<bool>(n, false)); //初始化
vector<pair<int,int>> border;
visit[row][col]=true;
dfs(grid,row,col,border,visit,s);
for(auto [x,y]:border){
grid[x][y]=color;
}
return grid;
}
void dfs(vector<vector<int>>& grid,int row,int col,vector<pair<int,int>>& border,vector<vector<bool>>& visit,int s){
int a[4][2]={{-1,0},{0,1},{1,0},{0,-1}};
int m=grid.size(),n=grid[0].size();
bool t=false;
for(int i=0;i<4;i++){
int nx=row+a[i][0],ny=col+a[i][1];
if(!(nx>=0&&nx<m&&ny>=0&&ny<n&&grid[nx][ny]==s)){
t=true;
}
else if(!visit[nx][ny]){
visit[nx][ny]=true;
dfs(grid,nx,ny,border,visit,s);
}
}
if(t){
border.push_back(pair<int,int>(row,col));
}
}
};