颜色填充(程序员面试金典08.10)Java深度优先遍历实现

70 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 4 天,点击查看活动详情

一、题目描述

编写函数,实现许多图片编辑软件都支持的「颜色填充」功能。

待填充的图像用二维数组 image 表示,元素为初始颜色值。初始坐标点的行坐标为 sr 列坐标为 sc。需要填充的新颜色为 newColor 。

「周围区域」是指颜色相同且在上、下、左、右四个方向上存在相连情况的若干元素。

请用新颜色填充初始坐标点的周围区域,并返回填充后的图像。

示例:

输入:

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] 内。

初始坐标点 (sr,sc) 满足 0 <= sr < image.length 和 0 <= sc < image[0].length 。

image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535] 内。

二、思路讲解

对于我这种算法小白来说,看见在矩阵里上下左右跑的题目,第一反应是使用dfs深度优先遍历 。

在题目给出的点上进行涂色操作,然后在他的上下左右分别找旧颜色,找到就继续涂色和寻找的操作,如果上面没找到就回溯,找下面,下面没找到,回溯,找左边……如果上下左右都没找到,证明已经涂色完成,返回涂好的数组。

三、java代码实现

class Solution {
    public static int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
		
        //保存原本的颜色
		int oldColor = image[sr][sc];
		
        //如果原本颜色就是要填充的颜色,直接将原来的image返回
		if(oldColor == newColor) {
			return image;
		}
		
		int m = image.length-1;
		int n = image[0].length-1;
 
        //将当前位置颜色改变
		image[sr][sc] = newColor;
		
		if(sr!=0 && image[sr-1][sc]==oldColor) {    //向上找旧颜色
			floodFill(image, sr-1, sc, newColor);
		}
		if(sr!=m && image[sr+1][sc]==oldColor) {    //向下找旧颜色
			floodFill(image, sr+1, sc, newColor);
		}
		if(sc!=0 && image[sr][sc-1]==oldColor) {    //向左找旧颜色
			floodFill(image, sr, sc-1, newColor);
		}
		if(sc!=n && image[sr][sc+1]==oldColor) {    //向右找旧颜色
			floodFill(image, sr, sc+1, newColor);
		}		
 
        //前后左右都没有旧颜色了,说明填充完了,执行回溯
		return image;	
    }	
}