包含整数的二维矩阵 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/image-smoother
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
注意:
- 给定矩阵中的整数范围为 [0, 255]。
- 矩阵的长和宽的范围均为 [1, 150]。
思路
遍历每一个元素,对每一个元素,遍历其周边的9个元素(带他本身),如果周边的元素有超界的就不加,不超界就加到新的矩阵对应的位置,并计数加1;依次类推,遍历完所有的元素后,求“平均值”。
代码
class Solution {
public:
vector<vector<int>> imageSmoother(vector<vector<int>>& M) {
int R=M.size(),C=M[0].size();
vector<int> v1(C,0);
vector<vector<int>> res(R,v1);
for(int r=0;r<R;r++)
{
for(int c=0;c<C;c++)
{
int count = 0;
for(int nr=r-1;nr<=r+1;nr++)
for(int nc=c-1;nc<=c+1;nc++)
{
if(nr>=0&&nr<R&&nc>=0&&nc<C){
res[r][c]+=M[nr][nc];
count++;
}
}
res[r][c]/=count;
}
}
return res;
}
};