Java 01矩阵leetcode

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

两个相邻元素间的距离为 1 。

class Solution {

    //static int [][] dirs = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};

    public int[][] updateMatrix(int[][] mat) {
        int n = mat.length,m = mat[0].length;
        //距离矩阵
        int [][] dist = new int[n][m];
        for(int i = 0;i < n;i++){
            Arrays.fill(dist[i],Integer.MAX_VALUE/2);
        }

        //如果（i,j）的元素为0，那么距离也为0
        for(int i = 0;i < n;i++){
            for(int j = 0;j < m;j++){
                if(mat[i][j] == 0){
                    dist[i][j] = 0;
                }
            }
        }

        //左上移动
        for(int i = 0;i < n;i++){
            for(int j = 0;j < m;j++){
                if(i - 1 >= 0){
                    dist[i][j] = Math.min(dist[i][j],dist[i - 1][j] + 1);
                }
                if(j - 1 >= 0 ){
                    dist[i][j] = Math.min(dist[i][j],dist[i][j - 1] + 1);
                }
            }
        }

        //右下移动
        for(int i = n-1;i>=0;i--){
            for(int j = m-1;j>=0;j--){
                if(i+1<n){
                    dist[i][j] = Math.min(dist[i][j],dist[i+1][j]+1);
                }
                if(j+1<m){
                    dist[i][j] = Math.min(dist[i][j],dist[i][j+1]+1);
                }
            }
        }
        return dist;
    }
}