题目:图像渲染
有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 0 到 65535 之间。
给你一个坐标 (sr, sc) 表示图像渲染开始的像素值(行 ,列)和一个新的颜色值 newColor,让你重新上色这幅图像。
为了完成上色工作,从初始坐标开始,记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
最后返回经过上色渲染后的图像。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/flood-fill 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例:
输入
:
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
/flood
-fill
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
想法:
递归函数和树形结构一直都是我的死穴,这次更是直接来了DFS算法,做题之前特地去看了一下题解,这篇主要是作为记录,答案来自于官方。 (啥时候对于递归问题能抬手就来,我就很满足了)
解法:
class Solution(object):
def floodFill(self
, image
, sr
, sc
, newColor
):
"""
:type image: List[List[int]]
:type sr: int
:type sc: int
:type newColor: int
:rtype: List[List[int]]
"""
color
= image
[sr
][sc
]
R
,C
= len(image
), len(image
[0])
if color
== newColor
: return image
def dfs(r
,c
):
if image
[r
][c
] == color
:
image
[r
][c
] = newColor
if r
>= 1: dfs
(r
-1,c
)
if r
+1 <R
:dfs
(r
+1,c
)
if c
>= 1: dfs
(r
,c
-1)
if c
+1 <C
:dfs
(r
,c
+1)
dfs
(sr
,sc
)
return image
效果: