面试题 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] 内。

执行用时:140 ms, 在所有 JavaScript 提交中击败了6.98%的用户

内存消耗:40.1 MB, 在所有 JavaScript 提交中击败了100.00%的用户

/**
 * @param {number[][]} image
 * @param {number} sr
 * @param {number} sc
 * @param {number} newColor
 * @return {number[][]}
 */
var floodFill = function (image, sr, sc, newColor) {
    if (image.length == 0 || image[sr][sc] == newColor) return image;  //数组为空 或者新颜色等于初始颜色就结束
    dfs(image, sr, sc, newColor, image[sr][sc]);
    return image;
};

var dfs = function (image, sr, sc, newColor, oldColor) {
    if (sr < 0 || sc < 0 || sr >= image.length || sc >= image[0].length) {
        return;
    }
    if (image[sr][sc] == oldColor && image[sr][sc] != newColor) {
        image[sr][sc] = newColor;
        dfs(image, sr + 1, sc, newColor, oldColor);
        dfs(image, sr - 1, sc, newColor, oldColor);
        dfs(image, sr, sc + 1, newColor, oldColor);
        dfs(image, sr, sc - 1, newColor, oldColor);
    }
}
经验分享 程序员 微信小程序 职场和发展