image-smoother.cpp

// Time:  O(m * n)
// Space: O(1)

class Solution {
public:
    vector<vector<int>> imageSmoother(vector<vector<int>>& M) {
        const auto& m = M.size(), &n = M[0].size();
        vector<vector<int>> result(M);
        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                result[i][j] = getGray(M, i, j);
            }
        }
        return result;
    }

private:
    int getGray(const vector<vector<int>>& M, int i, int j) {
        const auto& m = M.size(), &n = M[0].size();
        double total = 0.0;
        int count = 0;
        for (int r = -1; r < 2; ++r) {
            for (int c = -1; c < 2; ++c) {
                const auto& ii = i + r;
                const auto& jj = j + c;
                if (0 <= ii && ii < m && 0 <= jj && jj < n) {
                    total += M[ii][jj];
                    ++count;
                }
            }
        }
        return static_cast<int>(total / count);
    }
};