题目
有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 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,
因为它不是在上下左右四个方向上与初始点相连的像素点。
注意:
- image 和 image[0] 的长度在范围 [1, 50] 内。
- 给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
- image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。
答案
该题和岛屿数量问题类似,并且相对要简单一些
public class SolutionFloodFill {
public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
// 先记录一下需要替换的值
int oldColor = image[sr][sc];
// 如果oldColor与newColor相等,直接返回
if (oldColor == newColor) {
return image;
}
int xSize = image[0].length;
int ySize = image.length;
// 记录坐标的栈
Queue<int[]> coordinate = new LinkedList<>();
// 缓存已经处理的坐标
Set<String> cache = new HashSet<>();
coordinate.offer(new int[] {sr, sc});
while (!coordinate.isEmpty()) {
int[] cd = coordinate.poll();
int _sr = cd[0];
int _sc = cd[1];
if (cache.contains(_sr + "," + _sc)) {
// 已经处理,直接跳过
continue;
}
int imageVal = image[_sr][_sc];
if (imageVal != oldColor) {
continue;
}
cache.add(_sr + "," + _sc);
image[_sr][_sc] = newColor;
// 分别计算上下左右
// 右
if (_sc + 1 < xSize) {
coordinate.offer(new int[] {_sr, _sc + 1});
}
// 左
if (_sc - 1 >= 0) {
coordinate.offer(new int[] {_sr, _sc - 1});
}
// 下
if (_sr + 1 < ySize) {
coordinate.offer(new int[] {_sr + 1, _sc});
}
// 上
if (_sr - 1 >= 0) {
coordinate.offer(new int[] {_sr - 1, _sc});
}
}
return image;
}
public static void main(String[] args) {
SolutionFloodFill sff = new SolutionFloodFill();
// int[][] image = {{1, 1, 1}, {1, 1, 0}, {1, 0, 1}};
// int sr = 1;
// int sc = 1;
// int newColor = 2;
int[][] image = {{0, 1, 0}, {0, 0, 1}};
int sr = 1;
int sc = 1;
int newColor = 1;
int[][] ints = sff.floodFill(image, sr, sc, newColor);
for (int i = 0; i < ints.length; i++) {
String line = "";
for (int j = 0; j < ints[i].length; j++) {
line = line + ints[i][j] + " ";
}
System.out.println(line);
}
}
}
执行结果:
- 执行用时:11ms
- 内存消耗:41MB