题目来源于leetcode,解法和思路仅代表个人观点。传送门。
难度:简单
用时:00:10:00 (难得一遍过)
题目
有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 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]内。
思路
比较简单吧。就是BFS或者DFS的思路。我这里用了DFS。
需要注意的点:
- 需要将【原始颜色】保存起来。
- 当【原始颜色】和【新颜色】相同时,直接返回image。
代码
class Solution {
public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
int oriColor = image[sr][sc];
if(oriColor == newColor){
return image;
}
int[][] flag = new int[image.length][image[0].length];
dfs(image,sr,sc,oriColor,newColor,flag);
return image;
}
public void dfs(int[][] image,int r,int c,int oriColor,int newColor,int[][] flag){
//如果【越界】,【已经遍历过】,【有颜色阻挡】,直接返回
if(r < 0 || c < 0 || r >= image.length || c>= image[0].length || flag[r][c] == 1 || image[r][c] != oriColor){
return;
}
image[r][c] = newColor;
flag[r][c] = 1;
//上下左右
dfs(image,r-1,c,oriColor,newColor,flag);
dfs(image,r+1,c,oriColor,newColor,flag);
dfs(image,r,c-1,oriColor,newColor,flag);
dfs(image,r,c+1,oriColor,newColor,flag);
flag[r][c] = 1;
return;
}
}
算法复杂度
时间复杂度: O(n×m),其中 n 和 m 分别是二维数组的行数和列数。最坏情况下需要遍历所有的方格一次。
空间复杂度: O(n×m),其中 n 和 m 分别是二维数组的行数和列数。主要为栈空间的开销。

版权声明:本文为LittleSeedling原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。