难度:简单
20200816
有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 0 到 65535 之间。
给你一个坐标 (sr, sc) 表示图像渲染开始的像素值(行 ,列)和一个新的颜色值 newColor,让你重新上色这幅图像。
为了完成上色工作,从初始坐标开始,记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
最后返回经过上色渲染后的图像。
示例 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,
因为它不是在上下左右四个方向上与初始点相连的像素点。
dfs
class Solution {
private int[][] direction = {
{
-1, 0}, {
0, -1}, {
0, 1}, {
1, 0}};
boolean[][] used;
public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
if(image[sr][sc] == newColor){
return image;
}
int m = image.length;
int n = image[0].length;
used = new boolean[m][n];
dfs(image, sr, sc, newColor, image[sr][sc]);
return image;
}
public void dfs(int[][] image, int i, int j, int newColor, int origColor){
if(!inArea(image, i, j) || used[i][j] || image[i][j] != origColor){
return;
}
used[i][j] = true;
image[i][j] = newColor;
for(int k = 0; k < 4; k++){
int x = i + direction[k][0];
int y = j + direction[k][1];
dfs(image, x, y, newColor, origColor);
}
}
public boolean inArea(int[][] image, int i, int j){
return i >= 0 && i < image.length && j >= 0 && j < image[0].length;
}
}
欢迎关注公众号:GitKid,暂时每日分享LeetCode题解,在不断地学习中,公众号内容也在不断充实,欢迎扫码关注