给你一个大小为 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目大意:
本质上就是一个图的遍历问题,那么很显然有两种方法——bfs和dfs,在这里我先给出bfs解法。
实现思路:
新建一个二维数组,用来存储是否已经访问过该点,同时用一个容器存储是边界节点的点位,在最后将其替换为题目所给的color。而且由于是bfs,所以肯定要创建一个队列,来存储在该连通分量中的每个节点。
实现代码:
class Solution {
public:
typedef pair<int,int> pii;
vector<vector<int>> colorBorder(vector<vector<int>>& grid, int row, int col, int color) {
int dire[4][2]={{0,1},{1,0},{-1,0},{0,-1}};
int m=grid.size(),n=grid[0].size();
int ori=grid[row][col];
vector<vector<bool>> visited(m, vector<bool>(n, false));
visited[row][col]=true;
queue<pii> q;
vector<pii> border;
q.emplace(row,col);
while(!q.empty()){
pii & node=q.front();
q.pop();
int x=node.first;
int y=node.second;
bool isborder=false;
for(int i=0;i<4;i++){
int nx=x+dire[i][0];
int ny=y+dire[i][1];
if(nx<0||nx>=m||ny<0||ny>=n||grid[nx][ny]!=ori){
isborder=true;
}
else if(!visited[nx][ny]){
q.emplace(nx,ny);
visited[nx][ny]=true;
}
}
if(isborder) border.emplace_back(x,y);
}
for(auto [x,y]:border){
grid[x][y]=color;
}
return grid;
}
};
这里学到了一下几点:
1、首先是typedef pair<type,type> pii的应用。可以使队列等容器存储两个值。很实用。同时可以通过访问first和second来访问每对值中的任意一个。
2、对于vector来说,一次要加入一对值,必须用emplace_back()函数。
3、对于queue来说,一次要加入一对值,必须用emplace()函数。
4、还要注意这里第8行visited容器的初始化,其中等号右边的括号里第一个代表初始化元素的个数,第二个值代表每个元素的值。
5、注意一下这里增量数组的设置,这点稍微关注一下就可以了。
6、这里再讲一个我之前犯的错误,不要在过程中对颜色进行改动,以免干扰对是否为边界条件的判断。举个例子,如果说要改的颜色和所指定的点的颜色是一样的,在改动的过程中,其实你是不知不觉把连通分量的范围扩大了。这样得出来的答案大概率是错的。这点要注意,即,尽量做到,不要在程序中途对所要判断的量进行改变,这样比较稳妥。