力扣之图像渲染（python）

题目

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

你也被给予三个整数 sr ,  sc 和 newColor 。你应该从像素 image[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，因为它不是在上下左右四个方向上与初始点相连的像素点。

示例 2:

输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2
输出: [[2,2,2],[2,2,2]]

提示:

• m == image.length
• n == image[i].length
• 1 <= m, n <= 50
• 0 <= image[i][j], newColor < 216
• 0 <= sr < m
• 0 <= sc < n

思路

这是一个很典型的洪水填充，不会的同学可以去洪水填充算法详解-CSDN博客这里学习，这个博客讲的很透彻了

代码

class Solution(object):
    def floodFill(self, image, sr, sc, color):
        """
        :type image: List[List[int]]
        :type sr: int
        :type sc: int
        :type color: int
        :rtype: List[List[int]]
        """
        self.image = image
        self.oldcolor = image[sr][sc]
        self.dfs(sr,sc,color)
        return self.image

    def dfs(self, sr, sc, newcolor):
        if sr < 0 or sr >= len(self.image) or sc < 0 or sc >= len(self.image[0]) or self.image[sr][sc] != self.oldcolor or self.image[sr][sc] == newcolor:
            return
        self.image[sr][sc] = newcolor
        self.dfs(sr - 1, sc, newcolor)
        self.dfs(sr, sc - 1, newcolor)
        self.dfs(sr, sc + 1, newcolor)
        self.dfs(sr + 1, sc, newcolor)

结果