golang刷leetcode 技巧(27)颜色填充

2022-08-02 18:43:51 浏览数 (1)

颜色填充。编写函数,实现许多图片编辑软件都支持的“颜色填充”功能。给定一个屏幕(以二维数组表示,元素为颜色值)、一个点和一个新的颜色值,将新颜色值填入这个点的周围区域,直到原来的颜色值全都改变。

示例1:

输入:

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] 内。

给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。

image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。

解题思路:

1,一个点要不要填充除了检查是否越界外,还需要检查以下两个条件

A,染色如果和原色相同不染色

B,染色如果和选点颜色不一样不染色

2,如果满足染色条件,染色当前点,并递归染色周围点

3,这是一种深度优先的遍历方法

代码实现:

代码语言:javascript复制
func floodFill(image [][]int, sr int, sc int, newColor int) [][]int {
    dfs(image,sr,sc,image[sr][sc],newColor)
    return image
}

func dfs(image [][]int, sr int, sc int,val, newColor int){
    if sr<0 || sc<0||sr>=len(image) ||sc>=len(image[0]) ||image[sr][sc]==newColor || image[sr][sc]!=val{
        return
    }else{
        image[sr][sc]=newColor
    }

    dfs(image,sr 1,sc,val,newColor)
    dfs(image,sr,sc 1,val,newColor)
    dfs(image,sr-1,sc,val,newColor)
    dfs(image,sr,sc-1,val,newColor)
}

0 人点赞