面试题 08.10. 颜色填充-----深度优先搜索

题目表述

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

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

深度优先搜索

根据题目描述,只需要找起始节点四周相同颜色的块,然后将块染成新的颜色即可。递归的找可以被染色的块,用visited数组记录哪些块已经被访问。同理,普通的广度优先搜索也可以解该题。

class Solution {
    int[][] dirs = {{0,1}, {0,-1}, {1,0}, {-1,0}};
    public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        boolean[][] visited = new boolean[image.length][image[0].length];
                dfs(image, sr,sc,newColor,visited);

        return image;
    }
    public void dfs(int[][] image, int sr, int sc, int newColor,boolean[][] visited){
        if(sr < 0 || sc< 0 || sr >=image.length || sc >= image[0].length){
            return ;
        }
        if(visited[sr][sc]) return;
        visited[sr][sc] = true;
        for(int[] dir : dirs){
            int x = sr + dir[0];
            int y = sc + dir[1];
            if(x < 0 || y< 0 || x >=image.length || y >= image[0].length){
            continue;
        }
            if(image[x][y] != image[sr][sc]) continue;
            dfs(image, x, y, newColor,visited);
        }
        image[sr][sc] = newColor;

    }
}
posted @ 2022-06-04 21:47  YoungerWb  阅读(28)  评论(0编辑  收藏  举报