LeetCode 1034. 边框着色（BFS/DFS）
LeetCode 1034. 边框着色（BFS/DFS）
Michael阿明
发布于 2021-02-19 14:52:56
发布于 2021-02-19 14:52:56
文章被收录于专栏：Michael阿明学习之路
文章目录
1. 题目
给出一个二维整数网格 grid，网格中的每个值表示该位置处的网格块的颜色。
只有当两个网格块的颜色相同，而且在四个方向中任意一个方向上相邻时，它们属于同一连通分量。
连通分量的边界是指连通分量中的所有与不在分量中的正方形相邻（四个方向上）的所有正方形，或者在网格的边界上（第一行/列或最后一行/列）的所有正方形。
给出位于 (r0, c0) 的网格块和颜色 color，使用指定颜色 color 为所给网格块的连通分量的边界进行着色，并返回最终的网格 grid 。
示例 1：
输入：grid = [[1,1],[1,2]],
r0 = 0, c0 = 0, color = 3
输出：[[3, 3], [3, 2]]
示例 2：
输入：grid = [[1,2,2],[2,3,2]],
r0 = 0, c0 = 1, color = 3
输出：[[1, 3, 3], [2, 3, 3]]
示例 3：
输入：grid = [[1,1,1],[1,1,1],[1,1,1]],
r0 = 1, c0 = 1, color = 2
输出：[[2, 2, 2], [2, 1, 2], [2, 2, 2]]
提示：
1 <= grid.length <= 50
1 <= grid[0].length <= 50
1 <= grid[i][j] <= 1000
0 <= r0 < grid.length
0 <= c0 < grid[0].length
1 <= color <= 1000来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/coloring-a-border 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
2. 解题
- 简单的BFS/DFS即可
- 判断条件：周围点出界，或 周围的点颜色不同，就把当前点标记为边界
2.1 BFS
class Solution {
public:
vector<vector<int>> colorBorder(vector<vector<int>>& grid, int r0, int c0, int color) {
int m = grid.size(), n = grid[0].size(), i, j, k, x, y;
vector<vector<int>> dir = {{1,0},{0,1},{0,-1},{-1,0}};
int origin = grid[r0][c0];
vector<vector<bool>> vis(m, vector<bool>(n, false));
queue<pair<int,int>> q;
q.push(make_pair(r0, c0));
vis[r0][c0] = true;
while(!q.empty())
{
i = q.front().first;
j = q.front().second;
q.pop();
for(k = 0; k < 4; ++k)
{
x = i + dir[k][0];
y = j + dir[k][1];
if(x >=0 && x<m && y>=0 && y<n)//在界内
{
if(vis[x][y]) continue;//访问过了，下一个
if(grid[x][y] != origin)//没有访问，颜色不同
grid[i][j] = color;// i, j 旁边的 x, y跟它不一样，边界
else//没有访问，颜色一样，正常入队
{
q.push({x,y});
vis[x][y] = true;
}
}
else//出界了
grid[i][j] = color;//i,j 是边界
}
}
return grid;
}
};40 ms 13 MB
2.2 DFS
class Solution {
vector<vector<int>> dir = {{1,0},{0,1},{0,-1},{-1,0}};
int m, n, origin, col;
public:
vector<vector<int>> colorBorder(vector<vector<int>>& grid, int r0, int c0, int color) {
m = grid.size(), n = grid[0].size();
origin = grid[r0][c0], col = color;
vector<vector<bool>> vis(m, vector<bool>(n, false));
vis[r0][c0] = true;
dfs(grid,r0,c0,vis);
return grid;
}
void dfs(vector<vector<int>>& grid, int i, int j, vector<vector<bool>>& vis)
{
int x, y, k;
for(k = 0; k < 4; ++k)
{
x = i + dir[k][0];
y = j + dir[k][1];
if(x >=0 && x<m && y>=0 && y<n)//在界内
{
if(vis[x][y]) continue;//访问过了，下一个
if(grid[x][y] != origin)//没有访问，颜色不同
grid[i][j] = col;// i, j 旁边的 x, y跟它不一样，边界
else//没有访问，颜色一样
{
vis[x][y] = true;
dfs(grid, x, y, vis);
}
}
else//出界了
grid[i][j] = col;//i,j 是边界
}
}
};36 ms 12.9 MB
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2020/09/10 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录
腾讯云开发者
