661.图片平滑器

题目

661.图片平滑器

题目大意

图像平滑器 是大小为 3 x 3 的过滤器，用于对图像的每个单元格平滑处理，平滑处理后单元格的值为该单元格的平均灰度。

每个单元格的 平均灰度 定义为：该单元格自身及其周围的 8 个单元格的平均值，结果需向下取整。（即，需要计算蓝色平滑器中 9 个单元格的平均值）。

如果一个单元格周围存在单元格缺失的情况，则计算平均灰度时不考虑缺失的单元格（即，需要计算红色平滑器中 4 个单元格的平均值）。

给你一个表示图像灰度的 m x n 整数矩阵 img ，返回对图像的每个单元格平滑处理后的图像 。

样例

示例 1:

输入:img = [[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

示例 2:

输入: img = [[100,200,100],[200,50,200],[100,200,100]]
输出: [[137,141,137],[141,138,141],[137,141,137]]
解释:
对于点 (0,0), (0,2), (2,0), (2,2): floor((100+200+200+50)/4) = floor(137.5) = 137
对于点 (0,1), (1,0), (1,2), (2,1): floor((200+200+50+200+100+100)/6) = floor(141.666667) = 141
对于点 (1,1): floor((50+200+200+200+200+100+100+100+100)/9) = floor(138.888889) = 138

数据规模

提示:

• m == img.length
• n == img[i].length
• 1 <= m, n <= 200
• 0 <= img[i][j] <= 255

思路

直接依次计算每一个位置处理后的结果即可：统计它$3\times 3$方针中的所有有效元素，对于矩阵界外的元素不计算进去。枚举其周围的九个单元是否存在，对于存在的单元格，统计其数量$tot$和总和$sum$，答案就是$\lfloor \frac{sum}{tot}\rfloor$

代码

// short int long float double bool char string void
// array vector stack queue auto const operator
// class public private static friend extern 
// sizeof new delete return cout cin memset malloc
// relloc size length memset malloc relloc size length
// for while if else switch case continue break system
// endl reverse sort swap substr begin end iterator
// namespace include define NULL nullptr exit equals 
// index col row arr err left right ans res vec que sta
// state flag ch str max min default charray std
// maxn minn INT_MAX INT_MIN push_back insert
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int>PII;
typedef pair<int, string>PIS;
const int maxn=1e5+50;//注意修改大小
long long read(){long long x=0,f=1;char c=getchar();while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}while(isdigit(c)){x=x*10+c-'0';c=getchar();}return x*f;}
ll qpow(ll x,ll q,ll Mod){ll ans=1;while(q){if(q&1)ans=ans*x%Mod;q>>=1;x=(x*x)%Mod;}return ans%Mod;}

class Solution {
public:
    vector<vector<int>> imageSmoother(vector<vector<int>>& img) {
		int n=img.size(),m=img[0].size();
		vector<vector<int>>ans(n,vector<int>(m));
		for(int i=0;i<n;i++){
			for(int j=0;j<m;j++){
				int sum=img[i][j],tot=1;
				if(i-1>=0&&j-1>=0)sum+=img[i-1][j-1],tot++;
				if(i-1>=0)sum+=img[i-1][j],tot++;
				if(i-1>=0&&j+1<=m-1)sum+=img[i-1][j+1],tot++;
				if(j-1>=0)sum+=img[i][j-1],tot++;
				if(j+1<=m-1)sum+=img[i][j+1],tot++;
				if(i+1<=n-1&&j-1>=0)sum+=img[i+1][j-1],tot++;
				if(i+1<=n-1)sum+=img[i+1][j],tot++;
				if(i+1<=n-1&&j+1<=m-1)sum+=img[i+1][j+1],tot++;
				ans[i][j]=sum/tot;
			}
		}
		return ans;
    }
};

class Solution:
    def imageSmoother(self, img: List[List[int]]) -> List[List[int]]:
        n, m = len(img), len(img[0])
        ans = [[0]*m for i in range(n)]
        for i in range(n):
            for j in range(m):
                sum,tot=0,0
                for x in range(max(i - 1, 0), min(i + 2, n)):
                    for y in range(max(j - 1, 0), min(j + 2, m)):
                        sum += img[x][y]
                        tot += 1
                ans[i][j]=sum//tot
        return ans