01 矩阵

/* 
 * 题目思路: 
 * 使用Java Stream API处理和计算矩阵中各个元素到最近0的距离。 
 * 1. 使用Queue储存所有为0的点，将其余点设为无穷大（Integer.MAX_VALUE）。 
 * 2. BFS遍历矩阵，用Stream进行迭代和处理。 
 */ 
import java.util.*; 
import java.util.stream.*; 
 
public class Solution { 
    public int[][] updateMatrix(int[][] mat) { 
        int m = mat.length; 
        int n = mat[0].length; 
        int[][] dist = new int[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}); 
                dist[i][j] = 0; 
            } else { 
                dist[i][j] = Integer.MAX_VALUE; 
            } 
        })); 
        int[][] dirs = {{1, 0}, {0, 1}, {-1, 0}, {0, -1}}; 
        while (!queue.isEmpty()) { 
            int[] cell = queue.poll(); 
            Arrays.stream(dirs).forEach(dir -> { 
                int r = cell[0] + dir[0]; 
                int c = cell[1] + dir[1]; 
                if (r >= 0 && r < m && c >= 0 && c < n && dist[r][c] > dist[cell[0]][cell[1]] + 1) { 
                    queue.offer(new int[]{r, c}); 
                    dist[r][c] = dist[cell[0]][cell[1]] + 1; 
                } 
            }); 
        } 
        return dist; 
    } 
}