题目
有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 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)
解题思路
一般这类题可以使用广度优先搜索和深度优先搜索,得益于颜色的更改,我们无需特意建立visit数组以确定哪些元素是否被访问。
#广度优先搜索
class Solution:def floodFill(self, image: List[List[int]], sr: int, sc: int, newColor: int) -> List[List[int]]:if image[sr][sc]==newColor:return imageQ=[] #队列Q.append((sr,sc))temp=image[sr][sc]while Q!=[]:t=Q[0]del Q[0]image[t[0]][t[1]]=newColorif t[0]-1>-1 and image[t[0]-1][t[1]]==temp:Q.append((t[0]-1,t[1]))if t[1]-1>-1 and image[t[0]][t[1]-1]==temp:Q.append((t[0],t[1]-1))if t[0]+1<len(image) and image[t[0]+1][t[1]]==temp:Q.append((t[0]+1,t[1]))if t[1]+1<len(image[0]) and image[t[0]][t[1]+1]==temp:Q.append((t[0],t[1]+1))return image
#深度优先搜索
class Solution:def floodFill(self, image: List[List[int]], sr: int, sc: int, newColor: int) -> List[List[int]]:if image[sr][sc]==newColor:return imagetemp=image[sr][sc]def dfs(x,y):if image[x][y]==temp:image[x][y]=newColorif x-1>-1 and image[x-1][y]==temp:dfs(x-1,y)if y-1>-1 and image[x][y-1]==temp:dfs(x,y-1)if x+1<len(image) and image[x+1][y]==temp:dfs(x+1,y)if y+1<len(image[0]) and image[x][y+1]==temp:dfs(x,y+1)dfs(sr,sc)return image