leetcode 733. 图像渲染

112 阅读2分钟

733. 图像渲染

一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 0 到 65535 之间。

给你一个坐标 (sr, sc) 表示图像渲染开始的像素值(行 ,列)和一个新的颜色值 newColor,让你重新上色这幅图像。

为了完成上色工作,从初始坐标开始,记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。

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

输入: sr = 1, sc = 1, newColor = 2
image = [[1,1,1],
		 [1,1,0],
		 [1,0,1]]
输出: [[2,2,2],
	  [2,2,0],
	  [2,0,1]]
解析: 
在图像的正中间,(坐标(sr,sc)=(1,1)),在路径上所有符合条件的像素点的颜色都被更改成2。
注意,右下角的像素没有更改为2,因为它不是在上下左右四个方向上与初始点相连的像素点。
  • imageimage[0] 的长度在范围 [1, 50] 内。
  • 给出的初始点将满足 0 <= sr < image.length0 <= sc < image[0].length
  • image[i][j]newColor 表示的颜色值在范围 [0, 65535]内。
class Solution:  # 模板题,BFS或者DFS均可
    def floodFill(self, image: List[List[int]], sr: int, sc: int, newColor: int) -> List[List[int]]:
        m, n = len(image), len(image[0])
        around = ((1, 0), (0, 1), (-1, 0), (0, -1))
        oldColor = image[sr][sc]

        stack = [(sr, sc)]  # 创建栈放入根节点

        while stack:  # 进入循环放入邻居
            r, c = stack.pop()
            if oldColor != newColor:  # 根剪枝
                image[r][c] = newColor

                for x, y in around:
                    x, y = r + x, c + y
                    if 0 <= x < m and 0 <= y < n \
                            and image[x][y] == oldColor:  # 邻剪枝
                        image[x][y] = newColor
                        stack.append((x, y))
        return image