題目傳送

問題描述

給定一個由 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;
    }
}