leetcode 733. Flood Fill 图像渲染 DFS

leetcode 2020年3月 每日一题打卡

题目：
有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。给你一个坐标 (sr, sc) 表示图像渲染开始的像素值（行 ，列）和一个新的颜色值 newColor，让你重新上色这幅图像。为了完成上色工作，从初始坐标开始，记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。最后返回经过上色渲染后的图像。

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flood-fill

思路： python ：dfs深度优先搜索。

细节： 坑在于有可能新图的颜色与原来的颜色相同c==newColor，需要做一下区别处理。

代码：

class Solution(object):
    def floodFill(self, image, sr, sc, newColor):
        """
        :type image: List[List[int]]
        :type sr: int
        :type sc: int
        :type newColor: int
        :rtype: List[List[int]]
        """
        x=len(image)
        y=len(image[0])
        c=image[sr][sc]
        same=False
        if c==newColor:
            newColor=65535+1+c
            same=True
        def dfs(image,i,j):
            if i>=0 and i<x and j>=0 and j<y:
                if image[i][j]==c:
                    image[i][j]=newColor
                    dfs(image, i-1, j)
                    dfs(image, i+1, j)
                    dfs(image, i, j-1)
                    dfs(image, i, j+1)
            return 

        dfs(image,sr,sc)
        if same:
            for i in range(0,x):
                for j in range(0,y):
                    if image[i][j]==newColor:
                        image[i][j]=c
        return image

本博客为原创作品,欢迎指导,转载请说明出处,附上本文链接,谢谢