[LeetCode] 542. 01 矩阵

题目传送

问题描述

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

示例 1:
输入:

0 0 0
0 1 0
0 0 0

输出:

0 0 0
0 1 0
0 0 0

示例 2:
输入:

0 0 0
0 1 0
1 1 1

输出:

0 0 0
0 1 0
1 2 1

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

个人思路

题中说明元素的方向为四方向，所以咱们在计算距离时，应当计算曼哈顿距离
步骤：
遍历矩阵，若是值为 0，则直接赋距离为 0
不然调用计算曼哈顿距离的函数，以须要计算的点为中心，从曼哈顿距离为 1 开始一圈一圈的遍历，直到找到 0，则返回当前的曼哈顿距离

AC代码：

class Solution {
    private static int find(int[][] matrix, int i, int j){
        int len = Math.max(matrix.length, matrix[0].length);
        for (int k = 1; k < len; k++){
            int m = i - k, n = j, temp = k;
            while (temp > 0){
                if (!(m >= matrix.length || m < 0)) {
                    if (!(n >= matrix[0].length || n < 0)) {
                        if (matrix[m][n] == 0)
                            return k;
                    }
                }
                m++;
                n++;
                temp--;
            }
            temp = k;
            while (temp > 0){
                if (!(m >= matrix.length || m < 0)) {
                    if (!(n >= matrix[0].length || n < 0)) {
                        if (matrix[m][n] == 0)
                            return k;
                    }
                }
                m++;
                n--;
                temp--;
            }
            temp = k;
            while (temp > 0){
                if (!(m >= matrix.length || m < 0)) {
                    if (!(n >= matrix[0].length || n < 0)) {
                        if (matrix[m][n] == 0)
                            return k;
                    }
                }
                m--;
                n--;
                temp--;
            }
            temp = k;
            while (temp > 0){
                if (!(m >= matrix.length || m < 0)) {
                    if (!(n >= matrix[0].length || n < 0)) {
                        if (matrix[m][n] == 0)
                            return k;
                    }
                }
                m--;
                n++;
                temp--;
            }
        }
        return 0;
    }

    public int[][] updateMatrix(int[][] matrix) {
        int[][] distance = new int[matrix.length][matrix[0].length];
        for (int i = 0; i < matrix.length; i++){
            for (int j = 0; j < matrix[0].length; j++){
                if (matrix[i][j] == 0){
                    distance[i][j] = 0;
                } else {
                    distance[i][j] = find(matrix, i, j);
                }
            }
        }
        return distance;
    }
}