Leetcode773:图像渲染(深度遍历解法)

117 阅读1分钟

题目: 有一幅以 m x n 的二维整数数组表示的图画 image ,其中 image[i][j] 表示该图画的像素值大小。 ...

解法:

var floodFill = function(image, sr, sc, newColor) {
    const initialColor = image[sr][sc]
    function flood(image,sr,sc){
        if(sr<0 || sc<0 ||sr>= image.length||sc>= image[0].length || image[sr][sc] === -1 || image[sr][sc] !== initialColor){
            return
        }else{
            image[sr][sc] = -1
            flood(image,sr-1,sc)
            flood(image,sr+1,sc)
            flood(image,sr,sc-1)
            flood(image,sr,sc+1)
        }
    }
    flood(image,sr,sc)
    for(let i=0;i<image.length;i++){
        for(let j=0;j<image[0].length;j++){
            if(image[i][j] === -1){
                image[i][j] =newColor
            }
        }
    }
    return image

};

置为-1防止重复遍历,具有普适性,此外还可以略加修改,直接让image[sr][sc]等于newColor,同时修改第一个if判断,避免重复搜索陷入死循环。