Easy

题目描述

图像平滑器是一个大小为 3 x 3 的过滤器,可以应用于图像的每个单元格,通过对该单元格和周围八个单元格的平均值向下取整来实现(即蓝色平滑器中九个单元格的平均值)。如果一个单元格的一个或多个周围单元格不存在,我们不会在平均值中考虑它(即红色平滑器中四个单元格的平均值)。

给定一个 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):floor(3/4) = floor(0.75) = 0
对于点 (0,1), (1,0), (1,2), (2,1):floor(5/6) = floor(0.83333333) = 0
对于点 (1,1):floor(8/9) = floor(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

解题思路

这道题的核心思路是对图像矩阵中每个位置进行 3x3 邻域的平均值计算。

解题思路:

  1. 遍历每个位置:对于原矩阵中的每个位置 (i, j),我们需要计算其 3x3 邻域的平均值。

  2. 边界处理:关键在于处理边界情况。对于边界上的位置,其 3x3 邻域可能不完整,我们只计算存在的邻居的平均值。

  3. 计算邻域均值

    • 遍历以 (i, j) 为中心的 3x3 区域,即从 (i-1, j-1) 到 (i+1, j+1)
    • 检查每个邻居位置是否在矩阵边界内
    • 累加有效邻居的值,并记录有效邻居的数量
    • 计算平均值并向下取整
  4. 创建新矩阵:为了避免修改原矩阵影响后续计算,需要创建一个新的结果矩阵。

算法步骤:

  1. 创建与原矩阵相同大小的结果矩阵
  2. 对每个位置 (i, j):
    • 初始化总和 sum 和计数 count
    • 遍历 3x3 邻域 (dx, dy) ∈ {-1, 0, 1} × {-1, 0, 1}
    • 检查邻居位置 (i+dx, j+dy) 是否有效
    • 累加有效邻居值和计数
    • 计算 sum/count 的整数部分

代码实现

class Solution {
public:
    vector<vector<int>> imageSmoother(vector<vector<int>>& img) {
        int m = img.size();
        int n = img[0].size();
        vector<vector<int>> result(m, vector<int>(n, 0));
        
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                int sum = 0;
                int count = 0;
                
                // 遍历3x3邻域
                for (int di = -1; di <= 1; di++) {
                    for (int dj = -1; dj <= 1; dj++) {
                        int ni = i + di;
                        int nj = j + dj;
                        
                        // 检查边界
                        if (ni >= 0 && ni < m && nj >= 0 && nj < n) {
                            sum += img[ni][nj];
                            count++;
                        }
                    }
                }
                
                result[i][j] = sum / count;
            }
        }
        
        return result;
    }
};
class Solution:
    def imageSmoother(self, img: List[List[int]]) -> List[List[int]]:
        m, n = len(img), len(img[0])
        result = [[0] * n for _ in range(m)]
        
        for i in range(m):
            for j in range(n):
                sum_val = 0
                count = 0
                
                # 遍历3x3邻域
                for di in range(-1, 2):
                    for dj in range(-1, 2):
                        ni, nj = i + di, j + dj
                        
                        # 检查边界
                        if 0 <= ni < m and 0 <= nj < n:
                            sum_val += img[ni][nj]
                            count += 1
                
                result[i][j] = sum_val // count
        
        return result
public class Solution {
    public int[][] ImageSmoother(int[][] img) {
        int m = img.Length;
        int n = img[0].Length;
        int[][] result = new int[m][];
        
        for (int i = 0; i < m; i++) {
            result[i] = new int[n];
            for (int j = 0; j < n; j++) {
                int sum = 0;
                int count = 0;
                
                // 遍历3x3邻域
                for (int di = -1; di <= 1; di++) {
                    for (int dj = -1; dj <= 1; dj++) {
                        int ni = i + di;
                        int nj = j + dj;
                        
                        // 检查边界
                        if (ni >= 0 && ni < m && nj >= 0 && nj < n) {
                            sum += img[ni][nj];
                            count++;
                        }
                    }
                }
                
                result[i][j] = sum / count;
            }
        }
        
        return result;
    }
}
var imageSmoother = function(img) {
    const m = img.length;
    const n = img[0].length;
    const result = Array(m).fill(0).map(() => Array(n).fill(0));
    
    for (let i = 0; i < m; i++) {
        for (let j = 0; j < n; j++) {
            let sum = 0;
            let count = 0;
            
            // 遍历3x3邻域
            for (let di = -1; di <= 1; di++) {
                for (let dj = -1; dj <= 1; dj++) {
                    const ni = i + di;
                    const nj = j + dj;
                    
                    // 检查边界
                    if (ni >= 0 && ni < m && nj >= 0 && nj < n) {
                        sum += img[ni][nj];
                        count++;
                    }
                }
            }
            
            result[i][j] = Math.floor(sum / count);
        }
    }
    
    return result;
};

复杂度分析

复杂度类型分析
时间复杂度O(m × n) - 需要遍历每个位置,每个位置检查最多9个邻居
空间复杂度O(m × n) - 需要创建一个新的结果矩阵