LeetCode 542. 01 矩阵

1. 题目

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

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

示例 2: 
输入:

0 0 0
0 1 0
1 1 1
输出:

0 0 0
0 1 0
1 2 1

注意:

给定矩阵的元素个数不超过 10000。
给定矩阵中至少有一个元素是 0。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/01-matrix
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题思路

因为有四个方向，一次遍历，从(0,0)开始，每次比较它左边和上边的元素；一次遍历从(m-1,n-1)开始，比较它右边和下边的元素。每个位置取两次遍历最小值。

class Solution {

    public int[][] updateMatrix(int[][] matrix) {   
        int m = matrix.length;
        if(m == 0) return matrix;
        int n = matrix[0].length;
        int[][] res = new int[m][n];

        for (int i = 0; i < m; i++) {
            Arrays.fill(res[i], Integer.MAX_VALUE);
        }
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                if(matrix[i][j] == 0){
                    res[i][j] = 0;
                }
                else{
                    if(i>=1){
                        if(res[i-1][j]!=Integer.MAX_VALUE){
                            res[i][j] = Math.min(res[i][j],res[i-1][j]+1);
                        }
                    }
                    if(j>=1 ){
                        if(res[i][j-1]!=Integer.MAX_VALUE){
                            res[i][j] = Math.min(res[i][j],res[i][j-1]+1);
                        }
                    }
                }
            }
        }
        for(int i=m-1; i>=0; i--){
            for(int j=n-1; j>=0; j--){
                if(matrix[i][j] == 0){
                    res[i][j] = 0;
                }
                else{
                    if(i+1<m){
                        if(res[i+1][j]!=Integer.MAX_VALUE){
                            res[i][j] = Math.min(res[i][j],res[i+1][j]+1);
                        }
                    }
                    if(j+1<n){
                        if(res[i][j+1]!=Integer.MAX_VALUE){
                            res[i][j] = Math.min(res[i][j],res[i][j+1]+1);
                        }
                    }
                }
            }
        }
        return res;
    }
}