首先看P1034边界着色问题。
题目意思就是说给定一个位置,要求把这个位置和这个位置属于同一连通分量的网格块染成给定的颜色(color),是一道很经典的搜索问题,我们从给定位置出发开始搜索,可以用dfs以及bfs,本文都采用的是dfs方式。
搜索的方向我们用一个4×2的二维数组来决定,4个方向,每个方向代表x轴和y轴的变化量。
是否搜索过我们用visit二维数组来记录,起始值除了起始点都是false,搜索的过程中每搜索一个点就将此点置为true意为已经搜索过了。
在搜索过程中我们定义变量isBorder,起始值是false,当搜到这个点判断这个点的条件,若满足条件时我们将isBorder置为true,意为这是起始点的一个连通分量网格块,记录下来,最后将所有的记录的点的颜色都置为给定的颜色(color)即可。
最后是判断当搜到这个点是,这个点是否满足条件。需要满足的条件为:**这个点的下一个方向的点已经超过了边界或者下一个方向的点的颜色和起始点的颜色不同。**所以问题已经很简单了,实现代码如下:
void dfs(vector<vector<int>>&grid,int x,int y,vector<vector<bool>> &visit,vector<pair<int,int>>&borders,int originalColor){
int m=grid.size();
int n=grid[0].size();
int dir[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
bool isBorder=false;
for(int i=0;i<4;i++){
int nx=dir[i][0]+x;
int ny=dir[i][1]+y;
if(nx<0||nx>=m||ny<0||ny>=n||grid[nx][ny]!=originalColor){ //满足是边界连通块的条件
isBorder=true;
}
else if(!visit[nx][ny]){ //如果没访问过就访问他
visit[nx][ny]=true;
dfs(grid,nx,ny,visit,borders,originalColor);
}
}
if(isBorder){
borders.push_back(make_pair(x,y));
}
}
vector<vector<int>> colorBorder(vector<vector<int>>& grid, int row, int col, int color) {
int m=grid.size();
int n=grid[0].size();
vector<vector<bool>>visit(m,vector<bool>(n,false));
vector<pair<int,int>>borders; //记录所有的边界联通块的位置
int originalColor=grid[row][col]; //记录起始点的颜色
visit[row][col]=true; //从起始点开始访问
dfs(grid,row,col,visit,borders,originalColor);
for(int i=0;i<borders.size();i++){ //将所有记录的联通分量边界颜色置为color
int x=borders[i].first;
int y=borders[i].second;
grid[x][y]=color;
}
return grid;
}
其次是P200岛屿数量
这个题跟上一个题的区别就是:上一个题需要把所有的边界找出来,而这个题我们只需要判断我们进行了几次dfs操作就行了,同时将所有我们搜索过的结点记录一下防止再次搜索。本题采用的方法是直接修改grid数组,将1置为0,表示已经搜索过了,我们也可以使用visit数组记录搜索的过程,代码如下:
void dfs(vector<vector<char>> &grid,int x,int y){
grid[x][y]='0'; //直接修改grid数组,0表示已经访问过了
int m=grid.size();
int n=grid[0].size();
if(x-1>=0&&grid[x-1][y]=='1') dfs(grid,x-1,y);
if(x+1<m&&grid[x+1][y]=='1') dfs(grid,x+1,y); //注意:x+1指的是行数加1,相当于对y轴操作
if(y-1>=0&&grid[x][y-1]=='1') dfs(grid,x,y-1);
if(y+1<n&&grid[x][y+1]=='1') dfs(grid,x,y+1); //同理,y+1是列加一,相当于对x轴操作
}
int numIslands(vector<vector<char>>& grid) {
int m=grid.size();
if(m==0) return 0; //防止n溢出,一定要提前判断
int n=grid[0].size();
int ans=0;
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(grid[i][j]=='1') {
dfs(grid,i,j);
ans++; //记录一下进行了几次dfs操作
}
}
}
return ans;
}