颜色填充。编写函数,实现许多图片编辑软件都支持的“颜色填充”功能。
给定一个屏幕(以二维数组表示,元素为颜色值)、一个点和一个新的颜色值,将新颜色值填入这个点的周围区域,直到原来的颜色值全都改变。
示例1:
输入:
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]内。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/color-fill-lcci 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
int m = image.size(), n = image[0].size();
int original = image[sr][sc], k, x, y, x0, y0;
vector<vector<int>> dir = {{1,0},{0,1},{0,-1},{-1,0}};
queue<vector<int>> q;
vector<vector<bool>> visited(m, vector<bool>(n,false));
q.push({sr,sc});
visited[sr][sc] = true;
image[sr][sc] = newColor;
while(!q.empty())
{
x0 = q.front()[0];
y0 = q.front()[1];
q.pop();
for(k = 0; k < 4; ++k)
{
x = x0+dir[k][0];
y = y0+dir[k][1];
if(x>=0 && x<m && y>=0 && y<n && !visited[x][y] && image[x][y]==original)
{
q.push({x,y});
visited[x][y] = true;
image[x][y] = newColor;
}
}
}
return image;
}
};
class Solution {
vector<vector<int>> dir = {{1,0},{0,1},{0,-1},{-1,0}};
int m, n, original;
vector<vector<bool>> visited;
public:
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
m = image.size(), n = image[0].size();
original = image[sr][sc];
visited.resize(m, vector<bool>(n,false));
visited[sr][sc] = true;
image[sr][sc] = newColor;
dfs(image,sr,sc,newColor);
return image;
}
void dfs(vector<vector<int>>& image, int x0, int y0, int newColor)
{
int x, y;
for(int k = 0; k < 4; ++k)
{
x = x0+dir[k][0];
y = y0+dir[k][1];
if(x>=0 && x<m && y>=0 && y<n && !visited[x][y] && image[x][y]==original)
{
visited[x][y] = true;
image[x][y] = newColor;
dfs(image,x,y,newColor);
//占领即可,不必回溯
}
}
}
};