提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  IT要闻

当前栏目

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

程序员 面试
2023-03-14 22:52:15 时间

一、题目描述


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

待填充的图像用二维数组 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 ，因为它不属于初始坐标点的周围区域。


二、思路讲解 


对于我这种算法小白来说，看见在矩阵里上下左右跑的题目，第一反应是使用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;
        
    }
    
    
}


相关文章