【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深度优先搜索

待续