leetcode 1矩阵

作为做这道题做到有点想哭的我,现在心情复杂,原理使用曼哈顿距离,bfs斜向分析;代码以下,转自另一位博主代码,因为,c++与java有点不一样,而我并不知道如何将vector<vector<int>> martix转换作为返回值返回,所以我的内心是崩溃的,这个暂时留待之后解决,先这样。
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;
    }
}

猜你喜欢

转载自blog.csdn.net/sos768/article/details/83004876