此公众号会发表计算机考研(初复试信息)、夏令营等资料,方便考研人对信息的获取,节约自身查找资料的时间,回复408,可获得数据结构、操作系统、计算机网络、计算机组成原理全科资料
编写函数,实现许多图片编辑软件都支持的「颜色填充」功能。
待填充的图像用二维数组 image 表示,元素为初始颜色值。初始坐标点的横坐标为 sr 纵坐标为 sc。需要填充的新颜色为 newColor 。
「周围区域」是指颜色相同且在上、下、左、右四个方向上存在相连情况的若干元素。
请用新颜色填充初始坐标点的周围区域,并返回填充后的图像。
示例:
输入:
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 ,因为它不属于初始坐标点的周围区域。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/color-fill-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution:
def floodFill(self, image: List[List[int]], sr: int, sc: int, newColor: int) -> List[List[int]]:
m, n = len(image), len(image[0])
visited = [[False]*n for _ in range(m)]#设置一个m*n的标记数组
direction = [[-1, 0], [1, 0], [0, -1], [0, 1]]#向上下左右找
def dfs(i,j):
if image[i][j] == oldColor:
image[i][j] = newColor
for d in direction:
new_x = d[0] + i
new_y = d[1] + j
if 0 <= new_x < m and 0 <= new_y < n and not visited[new_x][new_y] and image[new_x][new_y] == oldColor:#满足条件依次往下找符合条件的
visited[new_x][new_y] = True
dfs(new_x,new_y)
oldColor = image[sr][sc]
dfs(sr,sc)
return image