刷题日记-733. 图像渲染

87 阅读2分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述

733. 图像渲染 - 力扣(LeetCode) (leetcode-cn.com)

有一幅以 m x n 的二维整数数组表示的图画 image ,其中 image[i][j] 表示该图画的像素值大小。

你也被给予三个整数 sr ,  sc 和 newColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。

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

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

 

示例 1:

image.png

输入: 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,因为它不是在上下左右四个方向上与初始点相连的像素点。

示例 2:

输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2 输出: [[2,2,2],[2,2,2]]  

提示:

m == image.length n == image[i].length 1 <= m, n <= 50 0 <= image[i][j], newColor < 216 0 <= sr < m 0 <= sc < n

二、思路分析

设置visited矩阵的时候,一定要注意先检测index是否合格(即是否在0~m-1这个范围内),然后再设置。不然负index会引起错误

三、AC 代码

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 = [[0]*n for _ in range(m)]

        def render_around(x, y, t):
            if x < 0 or y < 0 or x > m-1 or y > n-1 or visited[x][y]==1:
                return
            
            visited[x][y] = 1
            
            if image[x][y] == t:
                image[x][y] = newColor
                render_around(x-1, y, t)
                render_around(x+1, y, t)
                render_around(x, y-1, t)
                render_around(x, y+1, t)
                
        
        render_around(sr, sc, image[sr][sc])
        return image