leetcode_661 图片平滑器

66 阅读1分钟

要求

包含整数的二维矩阵 M 表示一个图片的灰度。你需要设计一个平滑器来让每一个单元的灰度成为平均灰度 (向下舍入) ,平均灰度的计算是周围的8个单元和它本身的值求平均,如果周围的单元格不足八个,则尽可能多的利用它们。

示例 1:

输入:
[[1,1,1],
 [1,0,1],
 [1,1,1]]
输出:
[[0, 0, 0],
 [0, 0, 0],
 [0, 0, 0]]
解释:
对于点 (0,0), (0,2), (2,0), (2,2): 平均(3/4) = 平均(0.75) = 0
对于点 (0,1), (1,0), (1,2), (2,1): 平均(5/6) = 平均(0.83333333) = 0
对于点 (1,1): 平均(8/9) = 平均(0.88888889) = 0

注意:

  • 给定矩阵中的整数范围为 [0, 255]。
  • 矩阵的长和宽的范围均为 [1, 150]。

核心代码

class Solution:
    def imageSmoother(self, img: List[List[int]]) -> List[List[int]]:
        height,width = len(img),len(img[0])
        res = [[None for _ in range(width)] for _ in range(height)]

        for h in range(height):
            for w in range(width):
                region_pixels = [pixel for row in img[max(0,h-1):h + 2] for pixel in row[max(0,w - 1):w + 2]]
                res[h][w] = sum(region_pixels) // len(region_pixels)
        return res

image.png

解题思路:其实就是针对二维数组中在一个点周围取值的过程,主要注意一点就是数组不要越界即可,注意这个地方中获取region_pixels的获取方式,值得借鉴。