有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 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,
因为它不是在上下左右四个方向上与初始点相连的像素点。
注意:
1、image 和 image[0] 的长度在范围 [1, 50] 内。
2、给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
3、image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。
思路:
与岛屿数量同类型,BFS和DFS都可,我用的BFS,但是犯了一个致命错误,没有把image[sr][sc]的值存储起来,导致image[sr][sc]被改变,致使题目出错。
代码:
class Solution {
public:
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
int color = image[sr][sc];
if(color==newColor)
return image;
queue<pair<int,int>> q;
int rsize=image.size();
int csize=image[0].size();
image[sr][sc] = newColor;
q.push({sr,sc});
while(!q.empty())
{
auto value=q.front();
q.pop();
int row=value.first;
int col=value.second;
if(row-1>=0&&image[row-1][col]==color)
{
q.push({row-1,col});
image[row-1][col]=newColor;
}
if(row+1<rsize&&image[row+1][col]==color)
{
q.push({row+1,col});
image[row+1][col]=newColor;
}
if(col-1>=0&&image[row][col-1]==color)
{
q.push({row,col-1});
image[row][col-1]=newColor;
}
if(col+1<csize&&image[row][col+1]==color)
{
q.push({row,col+1});
image[row][col+1]=newColor;
}
}
return image;
}
};