Leetcode 733. 图像渲染
题目
有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 0 到 65535 之间。
给你一个坐标 (sr, sc) 表示图像渲染开始的像素值(行 ,列)和一个新的颜色值 newColor,让你重新上色这幅图像。
为了完成上色工作,从初始坐标开始,记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
最后返回经过上色渲染后的图像。
示例:
输入:
image = [[1,1,1],[1,1,0],[1,0,1]]
sr = 1, sc = 1, newColor = 2
输出: [[2,2,2],[2,2,0],[2,0,1]]
解析:
在图像的正中间,(坐标(sr,sc)=(1,1)),
在路径上所有符合条件的像素点的颜色都被更改成2。
注意,右下角的像素没有更改为2,
因为它不是在上下左右四个方向上与初始点相连的像素点。
注意:
- image 和 image[0] 的长度在范围 [1, 50] 内。
- 给出的初始点将满足 0 <= sr < image.length 和0 <= sc < image[0].length。
- image[i][j] 和 newColor 表示的颜色值在范围 [0,65535]内。
题解
bfs或dfs
记录原颜色和新颜色,使用bfs或dfs进行颜色修改。
详细过程见代码
代码
- bfs
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
int m = image.size(),n = image[0].size();
queue<pair<int,int>> q;
q.push(make_pair(sr,sc));
int oldColor = image[sr][sc],size;
if(oldColor == newColor) return image; //颜色没有发生改变,不需要修改
int dir[4][2] = {{-1,0},{0,1},{1,0},{0,-1}};
while(!q.empty()){
size = q.size();
while(size--){
pair<int,int> now = q.front();
q.pop();
image[now.first][now.second] = newColor;
for(int i=0; i<4; i++){
if(now.first+dir[i][0]>=0 && now.first+dir[i][0]<m && now.second+dir[i][1]>=0 && now.second+dir[i][1]<n && image[now.first+dir[i][0]][now.second+dir[i][1]]==oldColor){
q.push(make_pair(now.first+dir[i][0],now.second+dir[i][1]));
}
}
}
}
return image;
}
- dfs
void dfs(int sr,int sc,int m,int n,vector<vector<int>>& image,int oldColor,int newColor){
if(sr==m || sr<0 || sc==n || sc<0 || image[sr][sc]!=oldColor) return;
image[sr][sc] = newColor;
dfs(sr+1,sc,m,n,image,oldColor,newColor);
dfs(sr,sc+1,m,n,image,oldColor,newColor);
dfs(sr-1,sc,m,n,image,oldColor,newColor);
dfs(sr,sc-1,m,n,image,oldColor,newColor);
}
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
int m = image.size(),n=image[0].size();
if(image[sr][sc] != newColor)
dfs(sr,sc,m,n,image,image[sr][sc],newColor);
return image;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/flood-fill
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。