题目
包含整数的二维矩阵 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
注意:
给定矩阵中的整数范围为 [0, 255]。
矩阵的长和宽的范围均为 [1, 150]。
解题
- 这个解题思路就很简单, 每个元素对周边八个元素加一起求和, 如果不满八个元素, 就有几个求几个, 注意边界即可, 思路很简单, 就是代码有些复杂, 毕竟要考虑周围的八个点, 代码如下 :
class Solution {
// row矩阵行数,col矩阵列数
// 对M[l][r]这个点求平均数
private int getAverage(int[][] M, int row, int col, int l, int r){
int sum = M[l][r];
int count = 1;
// 左上M[l-1][r-1]
if(l - 1 >= 0 && r -1 >= 0){
sum+=M[l-1][r-1];
count++;
}
// 上中M[l-1][r]
if(l - 1 >= 0){
sum+=M[l-1][r];
count++;
}
// 右上M[l-1][r+1]
if(l - 1 >= 0 && r + 1 < col){
sum+=M[l-1][r+1];
count++;
}
// 左中M[l][r-1]
if(r - 1 >= 0){
sum+=M[l][r-1];
count++;
}
// 右中M[l][r+1]
if(r + 1 < col){
sum+=M[l][r+1];
count++;
}
// 左下M[l+1][r-1]
if(l + 1 < row && r - 1 >= 0){
sum+=M[l+1][r-1];
count++;
}
// 下中M[l+1][r]
if(l + 1 < row){
sum+=M[l+1][r];
count++;
}
// 右下M[l + 1][r + 1]
if(l + 1 < row && r + 1 < col){
sum+=M[l+1][r + 1];
count++;
}
// int型除法小数会舍去
return sum / count;
}
public int[][] imageSmoother(int[][] M) {
int rows = M.length;
int cols = M[0].length;
int [][] newM = new int[rows][cols];
for(int i = 0; i < rows; i++){
for(int j = 0; j < cols; j++){
newM[i][j] = getAverage(M, rows, cols, i, j);
}
}
return newM;
}
}