墙与门

/*
 * 题目：墙与门
 * 你被给定一个 m × n 的二维网格，网格中有以下几种可能的初始化值：
 * -1 表示墙或是障碍物
 * 0 表示一扇门
 * INF 无限表示一个空的房间
 * 你需要填充每个空房间到最近门的距离，如果无法到达门，则填充 INF
 *
 * 思路：
 * 使用广度优先搜索（BFS）从每个门（值为0）开始，将距离依次填充到周围的空房间
 */
 
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
import java.util.stream.IntStream;
 
public class WallsAndGatesStream {
    private static final int INF = Integer.MAX_VALUE;
 
    public void wallsAndGates(int[][] rooms) {
        if (rooms == null || rooms.length == 0) return;
 
        int m = rooms.length;
        int n = rooms[0].length;
        Queue<int[]> queue = new LinkedList<>();
 
        // 将所有的门加入队列
        IntStream.range(0, m).forEach(i ->
            IntStream.range(0, n).filter(j -> rooms[i][j] == 0).forEach(j -> queue.offer(new int[]{i, j}))
        );
 
        // 四个方向
        int[][] directions = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
 
        // BFS 处理
        while (!queue.isEmpty()) {
            int[] point = queue.poll();
            int row = point[0];
            int col = point[1];
 
            Arrays.stream(directions).forEach(direction -> {
                int newRow = row + direction[0];
                int newCol = col + direction[1];
 
                // 检查边界和是否是空房间
                if (newRow >= 0 && newRow < m && newCol >= 0 && newCol < n && rooms[newRow][newCol] == INF) {
                    // 更新距离并将新的房间加入队列
                    rooms[newRow][newCol] = rooms[row][col] + 1;
                    queue.offer(new int[]{newRow, newCol});
                }
            });
        }
    }
}