661. 图片平滑器

 

 

class Solution(object):
    def imageSmoother(self, M):
        """
        :type M: List[List[int]]
        :rtype: List[List[int]]
        """
        # padding：在原矩阵周围加上一圈非有效数字：-10，便于处理边缘元素
        m = len(M[0])
        N = [[-10] + i + [-10] for i in M]
        N = [[-10] * (m + 2)] + N + [[-10] * (m + 2)]
        for i in range(1, len(N) - 1):
            for j in range(1, len(N[0]) - 1):
                # 当前元素及其周围的八个元素：九宫格
                total = [N[i - 1][j - 1], N[i - 1][j], N[i - 1][j + 1], N[i][j - 1], N[i][j], N[i][j + 1],
                         N[i + 1][j - 1], N[i + 1][j], N[i + 1][j + 1]]
                # 统计包括当前九宫格内元素的和
                sumthis = 0
                # 统计当前九宫格内非有效元素的个数，用于求均值
                numextra = 0
                # 求九宫格内有效元素的和
                for digit in total:
                    if digit != -10:
                        sumthis += digit
                    else:
                        numextra += 1
                # 求均值并赋给原矩阵对应元素
                M[i - 1][j - 1] = sumthis // (9 - numextra)
        return M


if __name__ == '__main__':
    solution = Solution()
    print(solution.imageSmoother([[1, 1, 1], [1, 0, 1], [1, 1, 1]]))