leetcode 661. 图片平滑器(python)

今天药忘吃喽~ 2022-05-19 06:05 142阅读 0赞

包含整数的二维矩阵 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

**注意:**

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

--------------------

class Solution:
        def imageSmoother(self, M):
            """
            :type M: List[List[int]]
            :rtype: List[List[int]]
            """
            result=[]
            m,n=len(M),len(M[0])
            for i in range(m):
                result.append([])
                for j in range(n):
                    value,count=M[i][j],1
                    if i-1>=0:
                        value += M[i-1][j]
                        count += 1
                        if j-1>=0:
                            value += M[i-1][j-1]
                            count += 1
                        if j+1<n:
                            value += M[i-1][j+1]
                            count += 1
                    if i+1<m:
                        value += M[i+1][j]
                        count += 1
                        if j-1>=0:
                            value += M[i+1][j-1]
                            count += 1
                        if j+1<n:
                            value += M[i+1][j+1]
                            count += 1
                    if j-1>=0:
                        value += M[i][j-1]
                        count += 1
                    if j+1<n:
                        value += M[i][j+1]
                        count += 1
                    result[i].append(value/count)
            return result