733. 图像渲染
有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 0 到 65535 之间。
给你一个坐标 (sr, sc) 表示图像渲染开始的像素值(行 ,列)和一个新的颜色值 newColor,让你重新上色这幅图像。
为了完成上色工作,从初始坐标开始,记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
最后返回经过上色渲染后的图像。
很简单的一道连通域类型问题,只要考虑新颜色如果等于旧颜色带来的无限递归问题即可,直接贴代码
class Solution:
def floodFill(self, image: List[List[int]], sr: int, sc: int, newColor: int) -> List[List[int]]:
if len(image)== 0 :
return image
lenx,leny = len(image),len(image[0])
color = image[sr][sc]
if(newColor == color):
return image
def dfs(x,y):
if (not 0<=x<lenx or not 0<=y<leny):
return
if(image[x][y] == color):
image[x][y] = newColor
dfs(x-1,y)
dfs(x+1,y)
dfs(x,y-1)
dfs(x,y+1)
dfs(sr,sc)
return image