Java剑指 Offer II 107. 矩阵中的距离（击败48.58%用户）

题目：

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

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

示例 :

 

输入：mat = [[0,0,0],[0,1,0],[0,0,0]]
输出：[[0,0,0],[0,1,0],[0,0,0]]

思路：

图+广度优先搜索

也可以动态规划

太折磨了

复杂度：

时间：双重循环O（nt）。

空间：O（t）。

代码：

 public int[][] updateMatrix(int[][] mat) {
        int m = mat.length;
        int n = mat[0].length;
        //队列用来存放矩阵中格子坐标
        Queue<int[]> queue = new LinkedList<>();
        //res以记录把每个格子离最近0距离
        int[][] res = new int[m][n];
        //先初始化这个res数组
        for(int i = 0;i<m;++i){
            for(int j =0;j<n;++j){
                //如果本身是0，那自然距离为0
                if(mat[i][j] == 0) {
                    //将所有0位置放入队列
                    queue.add(new int[]{i,j});
                    res[i][j] = 0;
                }
                //不是0，先初始化最大，再更新
                else res[i][j] = Integer.MAX_VALUE;
            }
        }
        
        int[][] helper = {{-1,0},{1,0},{0,1},{0,-1}};
        while(!queue.isEmpty()){
            //将队列中的0位挨个弹出
            int[] pos = queue.remove();
            //定义距离
            int dist = res[pos[0]][pos[1]];
            for(int[] dir : helper){
                int r = pos[0] + dir[0];
                int c = pos[1] + dir[1];
                if(r >= 0&& c>=0 && r< m&&c<n){
                    if(res[r][c] > dist+1){
                        res[r][c] = dist+1;
                        queue.add(new int[]{r,c});
                    }
                }
            }
        }
        return res;
    }