题目概述:
有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 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]内。
算法思路:
1. 深度优先搜索
- 从给定坐标开始DFS,遇到相同颜色则涂色,遇到不同色则跳过,遇到涂过的色跳过
代码实现:
public int[][] dfs(int[][] image, int row, int column, int newColor, int src){
int rowLength = image.length;
int columnLength = image[0].length;
if(row >= rowLength || column >= columnLength || row < 0 || column < 0 || image[row][column] != src || image[row][column] == newColor){
return image;
}
else{
int temp = image[row][column];
image[row][column] = newColor;
dfs(image,row - 1,column,newColor,temp);
dfs(image,row + 1,column,newColor,temp);
dfs(image,row,column + 1,newColor,temp);
dfs(image,row,column - 1,newColor,temp);
}
return image;
}
public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
if(image == null || image.length == 0){
return new int[][]{};
}
return dfs(image,sr,sc,newColor,image[sr][sc]);
}
复杂度分析:
- 时间复杂度:O(n×m),其中 n 和 m 分别是二维数组的行数和列数。最坏情况下需要遍历所有的方格一次。
- 空间复杂度:O(n×m),其中 n 和 m 分别是二维数组的行数和列数。主要为栈的开销。
2. 广度优先搜索
代码实现:
public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
int srcColor = image[sr][sc];
if(image == null || image.length == 0){
return new int[][]{};
}
if(srcColor == newColor){
return image;
}
int[] dx = {1,0,0,-1};
int[] dy = {0,1,-1,0};
int rowLength = image.length;
int colLength = image[0].length;
Queue<int[]> roundColor = new LinkedList<>();
roundColor.offer(new int[]{sr,sc});
image[sr][sc] = newColor;
while(!roundColor.isEmpty()){
int[] cur = roundColor.poll();
int row = cur[0];
int col = cur[1];
for(int i = 0; i < 4; i++){
int curRow = row + dx[i];
int curCol = col + dy[i];
if(curRow >= 0 && curRow < rowLength && curCol >= 0 && curCol < colLength && image[curRow][curCol] == srcColor){
roundColor.offer(new int[]{curRow,curCol});
image[curRow][curCol] = newColor;
}
}
}
return image;
}
复杂度分析:
- 时间复杂度:O(n×m),其中 n 和 m 分别是二维数组的行数和列数。最坏情况下需要遍历所有的方格一次。
- 空间复杂度:O(n×m),其中 n 和 m 分别是二维数组的行数和列数。主要为队列的开销。