面试题 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 ，因为它不属于初始坐标点的周围区域。

提示：

• image 和 image[0] 的长度均在范围 [1, 50] 内。
• 初始坐标点 (sr,sc) 满足 0 <= sr < image.length 和 0 <= sc < image[0].length 。
• image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535] 内。

题目链接

题目分析

这道题是flood fill算法的经典应用，我的写法是dfs，当然还有bfs的写法，dfs写法相对于bfs来说存储的对象少，速度会快一些，但是容易爆栈，当然这道题不会，而bfs的话就不会用爆栈的问题，使用dfs的思路就是从题目中给定的(sr, sc)出发，找到所有和image[sr][sc]颜色相同的点，将它们染成newColor就好了

解题代码

Python

class Solution:
    def floodFill(self, image: List[List[int]], sr: int, sc: int, newColor: int) -> List[List[int]]:
        n, m = len(image), len(image[0])
        dx, dy = [-1, 0, 1, 0], [0, -1, 0, 1]

        def dfs(image, x, y, oc, nc):
            image[x][y] = nc
            for i in range(4):
                nx, ny = x + dx[i], y + dy[i]
                if nx<0 or nx>=n or ny<0 or ny>=m or image[nx][ny]!=oc:
                    continue
                if image[nx][ny]==nc:
                    continue
                image[nx][ny] = nc
                dfs(image, nx, ny, oc, nc)
        dfs(image, sr, sc, image[sr][sc], newColor)
        return image

Java

class Solution {
    
    static int n, m;
    static int[] dx = {-1, 0, 1, 0};
    static int[] dy = {0, -1, 0, 1};

    private void dfs(int[][] image, int x, int y, int oc, int nc){
        
        image[x][y] = nc;
        for(int i=0; i<4; i++){
            int nx = x + dx[i];
            int ny = y + dy[i];
            if(nx<0||nx>=n||ny<0||ny>=m||image[nx][ny]!=oc) continue;
            if(image[nx][ny]==nc) continue;
            image[nx][ny] = nc;
            dfs(image, nx, ny, oc, nc);
        }
    }

    public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        n = image.length;
        m = image[0].length;

        dfs(image, sr, sc, image[sr][sc], newColor);
        return image;
    }
}