小白学习[leetcode]之[动态规划]542. 01 矩阵

题目的链接在这里: https://leetcode-cn.com/problems/01-matrix/

---

题目大意

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

两个相邻元素间的距离为 1 。

---

一、示意图

二、解题思路

动态规划

代码如下：

class Solution {
 public int[][] updateMatrix(int[][] mat) {
        int rows=mat.length;
        int cols=mat[0].length;
       int[][]result=new int[rows][cols];
       
        //这里的动态规划就是 按照之前的那个方法 分解成各个子问题 要么就是先遍历一遍 把他等于0的点都给赋值了
        //然后把不是0的点给设置成O 然后就是 剩下那些点就都不是等于0 的 那肯定就是 四个方位里最小的那个 然后加一
        //理论上是没错的 但是需要遍历吧 形成一个新的矩阵
        //可是边界都是没用四个方位的  那就分开讨论嘛
        //是不是甚至都不用重新赋值呀 就直接把原来的复制过来就行了  因为他本来就是0 还是0 的
        //看到网上的思路了 就是从左上到右下进行一次动态规划 然后再从右下到左上又进行一次
        //先是左上到右下的
        for(int i=0;i<rows;i++){
            for(int j=0;j<cols;j++){
                //在这里进行初始化好了
                result[i][j]=1000;
                if(mat[i][j]==0) {
                    result[i][j] = 0;
                }
                else{
                    //那就是等于1 需要重新找到他的邻边的最小值 然后进行加一操作
                    //应该是这样理解 因为是从左上到右下的 所以遍历更新的情况应该是 一行一行进行的 所以只需要把当前这个和他同行的
                    //前一个进行比较 （当他不在第一列的时候）  如果是第一列的话 就只能和他上一行同一列的比较了 因为他同行的下一个
                    //还没被更新到 所以是需要判断是和他前一个 [i j-1]比 还是和 前一列[i-1 j] 可是存在一些中间值是可以
                    //同时和 i-1 j 以及 i j-1进行判断的哎
                    //为什么只和一个比呢  哦 好像没有哦 i>0和j>0的 就会同时把这两个if都进行一遍
                    if(j>0){
                        //如果是第一行 并且不是第一个的话 那就需要进行比较了
                        result[i][j]=Math.min(result[i][j],result[i][j-1]+1);
                    }
                    if(i>0){
                        result[i][j]=Math.min(result[i][j],result[i-1][j]+1);
                    }
                }
            }
        }
        //再是从右下到左上
        for(int i=rows-1;i>=0;i--){
            for(int j=cols-1;j>=0;j--){
                //这里就不用赋初值了
                if(mat[i][j]==0) {
                    result[i][j] = 0;
                }
                else{
                    if(i<rows-1){
                        result[i][j]=Math.min(result[i][j],result[i+1][j]+1);
                    }
                    if(j<cols-1){
                        result[i][j]=Math.min(result[i][j],result[i][j+1]+1);
                    }
                }
            }
        }
        return result;
    }
}