Image Smoother LeetCode solution in C++

1
class Solution {
2
 public:
3
  vector<vector<int>> imageSmoother(vector<vector<int>> &img) {
4
    int row[] = {0, 0, 0, -1, 1, -1, 1, -1, 1};
5
    int col[] = {0, -1, 1, 0, 0, 1, -1, -1, 1};
6
    int m = img.size(), n = img[0].size();
7
    vector<vector<int>> ans(m, vector<int>(n, 0));
8
    for (int i = 0; i < m; i++) {
9
      for (int j = 0; j < n; j++) {
10
        int sum = 0, count = 0;
11
        for (int k = 0; k < 9; k++) {
12
          if (0 <= i + row[k] && i + row[k] < m && 0 <= j + col[k] && j + col[k] < n) {
13
            sum += img[i + row[k]][j + col[k]];
14
            count++;
15
          }
16
        }
17
        ans[i][j] = sum / count;
18
      }
19
    }
20
    return ans;
21
  }
22
};