661. 图片平滑器

Introduction

Question: 661. 图片平滑器

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

解法一

Analysis

也没啥难度,注意一下别越界就好了,虽然好几个循环写的略复杂,但是把中间求平均的过程封装一下就简单很多了。

Implement

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
vector<vector<int>> imageSmoother(vector<vector<int>>& M) {
// m x n
int m = M.size(), n = M[0].size();
vector<vector<int>> res(m, vector<int>(n));
for(int i = 0;i < m;i++) {
for(int j = 0; j < n; j++) {
int sum = 0, c = 0;
for(int ii = max(0, i - 1), iiend = min(m, i + 2); ii < iiend; ii++) {
for(int jj = max(0, j - 1), jjend = min(n, j + 2); jj < jjend; jj++) {
sum += M[ii][jj];
c++;
}
}
res[i][j] = sum / c;
}
}
return res;
}