01矩阵

题目描述

给定一个由 0 和 1 组成的矩阵，找出每个元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。

思路

用到了BFS 可以和79单词搜索参考学习
BFS流程：（BFS类似树的层次遍历，需要一个队列，而DFS需要的是一个递归）

1. 任取一个顶点访问，入队，并将其标记为已读
2. 当队列不为空的时候：出队，依次检查出队顶点的邻接顶点，访问没有访问过的顶点，并将其入队
3. 当队为空的时候跳出循环，BFS结束

题解

class O1Matrix542 {
    public static void main(String[] args) {
        O1Matrix542 matrix542 = new O1Matrix542();
        int arr [][] = {{0, 0, 0}, {0, 1, 0}, {1, 1, 1}};
        int[][] res = matrix542.updateMatrix(arr);
        for (int i = 0; i < res.length; i++) {
            for (int j = 0; j < res[i].length; j++) {
//                System.out.print(arr[i][j]);
                System.out.format("%d ",res[i][j]);
            }
            System.out.println();
        }

    }
    public int[][] updateMatrix(int[][] matrix) {
        // 首先将所有的 0 都入队，并且将 1 的位置设置成 -1，表示该位置是 未被访问过的 1
        Queue<int[]> queue = new LinkedList<>();
        int m = matrix.length, n = matrix[0].length;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    queue.offer(new int[] {i, j});          //offer添加到队尾
                } else {
                    matrix[i][j] = -1;
                }
            }
        }

        int[] dx = new int[] {-1, 1, 0, 0};
        int[] dy = new int[] {0, 0, -1, 1};
        while (!queue.isEmpty()) {
            int[] point = queue.poll();
            int x = point[0], y = point[1];
            for (int i = 0; i < 4; i++) {
                int newX = x + dx[i];
                int newY = y + dy[i];
                // 如果四邻域的点是 -1，表示这个点是未被访问过的 1
                // 所以这个点到 0 的距离就可以更新成 matrix[x][y] + 1。
                if (newX >= 0 && newX < m && newY >= 0 && newY < n
                        && matrix[newX][newY] == -1) {
                    matrix[newX][newY] = matrix[x][y] + 1;
                    queue.offer(new int[] {newX, newY});
                }
            }
        }

        return matrix;
    }
}```