题目描述

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

样例

输入:
[[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]。

算法

(模拟) $O(nm)$

1. 没啥算法可言，直接按照题意做即可。可以定义两个偏移量数组来简化代码。

时间复杂度

• 每个格子遍历常数次，故时间复杂度为 $O(nm)$。

C++ 代码

class Solution {
public:
    vector<vector<int>> imageSmoother(vector<vector<int>>& M) {
        int n = M.size(), m = M[0].size();
        vector<vector<int>> ans(M);
        int dx[9] = {-1, -1, -1, 0, 1, 1, 1, 0, 0};
        int dy[9] = {-1, 0, 1, 1, 1, 0, -1, -1, 0};
        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++) {
                int tot = 0, cnt = 0;
                for (int k = 0; k < 9; k++) {
                    int x = i + dx[k], y = j + dy[k];
                    if (x >= 0 && x < n && y >= 0 && y < m) {
                        tot += M[x][y];
                        cnt++;
                    }
                }
                ans[i][j] = tot / cnt;
            }
        return ans;
    }
};