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

【思路】:
用BFS广搜。
从起点开始向四个方向泛洪。只要发现有目标颜色的点,就把他替换成newColor


【代码】:

/**
 * @param {number[][]} image
 * @param {number} sr
 * @param {number} sc
 * @param {number} newColor
 * @return {number[][]}
 */

var ishefa = function(image, x, y, color,vis){
    let m = image.length;
    let n = image[0].length;
    if(x < 0 || y < 0)  return false;
    if(x >= m || y >= n)    return false;
    if(image[x][y] != color)    return false;
    if(vis[x][y] == 1)  return false;	//防止重复加入队列
    return true;
}

var guangsou = function(image, sr, sc, newColor, vis){
    var que = [];
    que.push({x:sr, y:sc});
    var color = image[sr][sc];      //这个颜色的节点都要被替换
    while(que.length != 0){
        let front = que[0];
        console.log(front.x, front.y)
        vis[front.x][front.y] = 1;
        //上方
        if(ishefa(image, front.x - 1, front.y, color,vis)){
            que.push({x: front.x - 1, y: front.y});
        }
        //下
        if(ishefa(image, front.x + 1, front.y, color,vis)){
            que.push({x:front.x + 1, y:front.y});
        }
        //左
        if(ishefa(image, front.x, front.y - 1, color,vis)){
            que.push({x:front.x, y:front.y - 1});
        }
        //右
        if(ishefa(image, front.x, front.y + 1, color,vis)){
            que.push({x:front.x, y:front.y + 1});
        }
        image[front.x][front.y] = newColor;
        //队列第一个节点删掉
        que.shift();
    }
}

var floodFill = function(image, sr, sc, newColor) {
    var vis = Array(image.length).fill().map(()=> Array(image[0].length).fill(0))
    guangsou(image, sr, sc, newColor, vis);
    return image;
};

你可能感兴趣的:(BFS,LeetCode,算法,leetcode)