542. 01 矩阵

算法记录

LeetCode 题目：

  给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。

---

---

说明

一、题目

  给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。

二、分析

• 对于每一个 1 到 0 找最短路径, 也就可以抽象为从 0 开始遍历, 一直去寻找 1 并且更新能到达当前点的距离信息.
• 和迷宫问题相似, 只需要将剪枝的条件隔离出来,防止陷入死循环就可以了.
• 按照层次进行递归就可以拿到最短的信息, 也可以生成一条对应的问题状态求解树.

class Solution {
    int[][] direction = {{0,1},{0,-1},{1,0},{-1,0}};
    public int[][] updateMatrix(int[][] mat) {
        int[][] visit = new int[mat.length][mat[0].length];
        Queue<Data> queue = new ArrayDeque();
        init_data(queue, visit, mat);
        while(!queue.isEmpty()) {
            Data data = queue.poll();
            for(int i = 0; i < 4; i++) {
                int x = data.x + direction[i][0];
                int y = data.y + direction[i][1];
                if(x < 0 || y < 0 || x >= mat.length || y >= mat[0].length) continue;
                if(visit[x][y] != -1) continue;
                visit[x][y] = data.distance + 1;
                queue.add(new Data(x, y, data.distance + 1));
            }
        }
        return visit;
    }

    private void init_data(Queue queue,int[][] visit, int[][] mat) {
        for(int i = 0; i < mat.length; i++) {
            for(int j = 0; j < mat[0].length; j++) {
                if(mat[i][j] == 0) {
                    queue.add(new Data(i, j, 0));
                    visit[i][j] = 0;
                } else {
                    visit[i][j] = -1;
                }
            }
        }
    }
    class Data{
        int x;
        int y;
        int distance;
        public Data(int x, int y, int distance) {
            this.x = x;
            this.y = y;
            this.distance = distance;
        }
    }
}

---

总结

熟悉状态更新的计算方式。