经典老模板题了,直接上代码
对了这里有个小坑,就是可能出现image[sr][sc]已经是newcolor,所以我们需要先判断一下再搜索
class Solution {
public:
int nextt[4][2]={{0,1},{1,0},{0,-1},{-1,0}},tx,ty;
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
if(newColor==image[sr][sc])
return image;
dfs(image,sr,sc,newColor,image[sr][sc]);
return image;
}
void dfs(vector<vector<int>>& image, int x, int y, int newColor,int oldcolor)
{
if(image[x][y]==oldcolor)
{
image[x][y]=newColor;
for(int i=0;i<4;i++)
{
tx=x+nextt[i][0];
ty=y+nextt[i][1];
if(tx<0||ty<0||tx>=image.size()||ty>=image[0].size())
continue;
dfs(image,tx,ty,newColor,oldcolor);
}
}
return ;
}
};
下面这个是题解(和我写法不同而已)
class Solution {
public:
int n, m, w;
void DFS(vector<vector<int>>& image, int x, int y, int c){
if(x<0 || x>=n || y<0 || y>=m || image[x][y]!=w || image[x][y]==c) // 退出条件:位置越界 或 当前像素与起始点不同 或 当前像素与新颜色相同
return;
image[x][y] = c; // 渲染当前位置像素为新颜色值
DFS(image, x+1, y, c); // 深搜4个方向
DFS(image, x-1, y, c);
DFS(image, x, y+1, c);
DFS(image, x, y-1, c);
}
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
n=image.size();
m=image[0].size();
w=image[sr][sc];
if(w != newColor) // 起点像素与新颜色值不同则以此为起点开始深搜
DFS(image, sr, sc, newColor);
return image;
}
};