图片平滑器

/*
题目思路：
1. 使用Java Stream流来处理二维数组。
2. 对于每个单元格，获取其邻域的值，并计算平均值。
3. 使用嵌套的Stream来迭代邻域，并对结果进行汇总。
*/
 
import java.util.stream.IntStream;
 
public class ImageSmootherStream {
    public int[][] imageSmoother(int[][] img) {
        int m = img.length;
        int n = img[0].length;
        return IntStream.range(0, m).mapToObj(i ->
            IntStream.range(0, n).map(j -> {
                int[] directions = {-1, 0, 1};
                int[] sumAndCount = {0, 0};
                for (int di : directions) {
                    for (int dj : directions) {
                        int ni = i + di;
                        int nj = j + dj;
                        if (ni >= 0 && ni < m && nj >= 0 && nj < n) {
                            sumAndCount[0] += img[ni][nj];
                            sumAndCount[1]++;
                        }
                    }
                }
                return sumAndCount[0] / sumAndCount[1];
            }).toArray()
        ).toArray(int[][]::new);
    }
}