【Java】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 1 1

注意:

給定矩陣的元素個數不超過 10000。
給定矩陣中至少有一個元素是 0。
矩陣中的元素只在四個方向上相鄰: 上、下、左、右。

解題思路

1、DP動態規劃法

package com.company;

/**
 * DP（動態規劃）
 * 對於每個點，值爲0距離爲0，
 * 值不爲0，則值爲上下左右點最小的那個加1
 */
public class updateMatrix {

    public int[][] updateMatrix(int[][] matrix) {
        int row = matrix.length;
        int col = matrix[0].length;
        // 第一遍遍歷，從左上到右下，比較上面和左邊中較小的值
        // matrix[i][j]代表第(i,j)個點距離0最短的距離
        // 毫無疑問，所有的0的位置一定都是0
        // 所有1的位置初始化爲一個大於1的數(便於後面比較)
        // dp[i][j] = min(dp[i][j],min(dp[i-1][j],dp[i][j-1])+1)
        for(int i = 0;i < row;i++){
            for(int j = 0;j < col;j++){
                if(matrix[i][j] == 1){
                    matrix[i][j] = row + col;
                }
                if(i > 0){
                    matrix[i][j] = Math.min(matrix[i][j],matrix[i-1][j] + 1);
                }
                if(j > 0){
                    matrix[i][j] = Math.min(matrix[i][j], matrix[i][j - 1] + 1);
                }
            }
        }
        // 第二次遍歷，反向遍歷，根據相鄰右元素和下元素及當前元素的結果得出最終結果
        for (int i = row - 1; i >= 0; i--) {
            for (int j = col - 1; j >= 0; j--) {
                if (i < row - 1) {
                    matrix[i][j] = Math.min(matrix[i][j], matrix[i + 1][j] + 1);
                }
                if (j < col - 1) {
                    matrix[i][j] = Math.min(matrix[i][j], matrix[i][j + 1] + 1);
                }
            }
        }
        return matrix;
    }
}

2、BFS深度優先搜索

待續