地图中的最高点

/*
 * 思路：
 * 1. 使用Java Stream API，我们可以简化输入矩阵的处理。
 * 2. 我们仍然使用广度优先搜索（BFS）来确定每个格子的高度。
 * 3. 使用Stream API初始化队列和高度矩阵。
 */

import java.util.*;
import java.util.stream.*;

public class Solution {
    public int[][] highestPeak(int[][] isWater) {
        int m = isWater.length;
        int n = isWater[0].length;
        int[][] height = new int[m][n];
        boolean[][] visited = new boolean[m][n];
        Queue<int[]> queue = new LinkedList<>();

        // 使用Stream API初始化队列和已访问矩阵
        IntStream.range(0, m).forEach(i ->
            IntStream.range(0, n).forEach(j -> {
                if (isWater[i][j] == 1) {
                    queue.add(new int[]{i, j});
                    visited[i][j] = true;
                }
            })
        );

        // 四个方向的偏移量
        int[][] directions = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};

        // 开始BFS
        while (!queue.isEmpty()) {
            int[] cell = queue.poll();
            int x = cell[0];
            int y = cell[1];

            for (int[] d : directions) {
                int nx = x + d[0];
                int ny = y + d[1];

                if (nx >= 0 && ny >= 0 && nx < m && ny < n && !visited[nx][ny]) {
                    height[nx][ny] = height[x][y] + 1;
                    queue.add(new int[]{nx, ny});
                    visited[nx][ny] = true;
                }
            }
        }

        return height;
    }
}