01 矩阵

/*
题目思路：
1. 创建一个结果矩阵result，用于存储每个位置到最近的0的距离。
2. 使用两个队列分别存储当前层和下一层的元素，初始时将所有的0的位置加入队列。
3. 对队列中的每个元素进行广度优先搜索（BFS），更新其邻居的距离，如果邻居的距离更新，加入下一层队列。
4. 循环进行BFS，直到队列为空。
*/
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
import java.util.stream.IntStream;

public class Solution {
    public int[][] updateMatrix(int[][] mat) {
        int m = mat.length, n = mat[0].length;
        int[][] result = new int[m][n];
        boolean[][] visited = new boolean[m][n];
        Queue<int[]> queue = new LinkedList<>();

        IntStream.range(0, m).forEach(i -> IntStream.range(0, n).forEach(j -> {
            if (mat[i][j] == 0) {
                queue.offer(new int[]{i, j});
                visited[i][j] = true;
            }
        }));

        int[] directions = {-1, 0, 1, 0, -1};

        while (!queue.isEmpty()) {
            int[] cell = queue.poll();
            int x = cell[0], y = cell[1];
            for (int k = 0; k < 4; k++) {
                int newX = x + directions[k];
                int newY = y + directions[k + 1];
                if (newX >= 0 && newX < m && newY >= 0 && newY < n && !visited[newX][newY]) {
                    result[newX][newY] = result[x][y] + 1;
                    queue.offer(new int[]{newX, newY});
                    visited[newX][newY] = true;
                }
            }
        }

        return result;
    }
}