LeetCode 661. Image Smoother

Given a 2D integer matrix M representing the gray scale of an image, you need to design a smoother to make the gray scale of each cell becomes the average gray scale (rounding down) of all the 8 surrounding cells and itself. If a cell has less than 8 surrounding cells, then use as many as you can.

Example 1:

Input:
[[1,1,1],
 [1,0,1],
 [1,1,1]]
Output:
[[0, 0, 0],
 [0, 0, 0],
 [0, 0, 0]]
Explanation:
For the point (0,0), (0,2), (2,0), (2,2): floor(3/4) = floor(0.75) = 0
For the point (0,1), (1,0), (1,2), (2,1): floor(5/6) = floor(0.83333333) = 0
For the point (1,1): floor(8/9) = floor(0.88888889) = 0

Note:

  • The value in the given matrix is in the range of [0, 255].
  • The length and width of the given matrix are in the range of [1, 150].
class Solution {
public:
    bool islegal(int i, int j){
        if(i>=0&&i<row&&j>=0&&j<col)
           return true;
        else
           return false;
    }
    vector<vector<int>> imageSmoother(vector<vector<int>>& M) {
         vector<vector<int>> temp=M;
         row=M.size(), col=M[0].size();
         for(int i=0; i<row; i++)
             for(int j=0; j<col; j++){
                 double cnt=0,sum=0;
                 for(int k=0;k<9;k++){
                     int r=i+dict[k][0], c=j+dict[k][1];
                     if(islegal(r,c)){
                         cnt++;
                         sum+=M[r][c];
                     }
                 }
                 temp[i][j]=floor(sum/cnt);
             }
        return temp;
    }
private:
    int row, col;
    vector<vector<int>> dict={{1,0},{0,1},{1,1},{0,0},{-1,0},{0,-1},{-1,-1},{1,-1},{-1,1}};
};
posted @ 2018-12-06 22:14  A-Little-Nut  阅读(85)  评论(0编辑  收藏  举报