题目

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

最后返回经过上色渲染后的图像。

思路

使用深度优先搜索策略遍历所有点。
已经访问过的点使用visited,一个bool类型的矩阵

  1. class Solution:
  2. def floodFill(self, image: List[List[int]], sr: int, sc: int, newColor: int) -> List[List[int]]:
  3. if len(image) == 0 or len(image[0]) == 0: return image
  4. rows, cols = len(image), len(image[0])
  5. initColor = image[sr][sc]
  6. def dfs(image, sr, sc):
  7. if sr < 0 or sr >= rows or sc < 0 or sc >= cols:
  8. return
  9. if image[sr][sc] != initColor or image[sr][sc] == newColor:
  10. return
  11. image[sr][sc] = newColor
  12. # 深搜四个方向
  13. dfs(image, sr - 1, sc)
  14. dfs(image, sr + 1, sc)
  15. dfs(image, sr, sc - 1)
  16. dfs(image, sr, sc + 1)
  17. dfs(image, sr, sc)
  18. return image