01矩阵

01矩阵一个充满着坑的dp问题,今天我们用dfs来解决这个问题

1.首先遍历数组当1的左上右下四个方向没有0时我们给它初始化一个较大的值数组的行数加列数这样做是为了保证在我们寻路时确保此位通过+1不能遇见0

2.我们开始遍历这个二维数组当matrix[i][j]==1时开始递归逻辑

3.我们定义一个direction数组这个数组定义左上右下四个方向的移动坐标

4.在递归函数里开始循环direction newX=x+direction[k][0] newY=y+direction[k][1],当坐标合法并且此坐标的值大于等于x,y坐标的值+1那么替换newX,newY坐标的值,然后继续递归,这个比较就彰显了我们将1一步不能到0初始化为较大值的作用,这样的替换保证了找到的路径最小

举个例子:

如当前数组

0到0的距离为0不用管

当遇到第一个1时周围有0直接递归函数(不满足matrix[newX][newY]>matrix[x][y]+1)结束,不用递归当前位仍是1

当遇到第二个1同上

当遇到第四个1同上

第三个值不为1当遇到的时候此值>1+1 =2将此值替换为2

结果如下

上代码:

class Solution {
    private int[][] direction={{0,-1},{-1,0},{0,1},{1,0}};
    public int[][] updateMatrix(int[][] matrix) {
        //将1周围没有0的地方初始化为比较大的值
        for(int i=0;i<matrix.length;i++){
            for(int j=0;j<matrix[i].length;j++){
                if(matrix[i][j]==1&&!oneStep(matrix,i,j)){
                    matrix[i][j]=matrix.length+matrix[i].length;
                }
                }
            }
        //当为1时开始寻找零
        for(int i=0;i<matrix.length;i++){
            for(int j=0;j<matrix[i].length;j++){
                if(matrix[i][j]==1){
                    dfs(matrix,i,j);
                }
            }
        }
        return matrix;
    }
    //深度优先遍历
    public void dfs(int[][] matrix,int x,int y){
        for(int i=0;i<4;i++){
          int newX=x+direction[i][0];
          int newY=y+direction[i][1];
          if(isArea(matrix,newX,newY)&&matrix[x][y]+1<matrix[newX][newY]){
              matrix[newX][newY]=matrix[x][y]+1;
               dfs(matrix,newX,newY);
          }
        }
    }
    //1周围没有0的话把此位置为较大值
    public boolean oneStep(int[][] matrix,int x,int y){
                for(int k=0;k<4;k++){
                    int newX=x+direction[k][0];
                    int newY=y+direction[k][1];
                    if(isArea(matrix,newX,newY)&&matrix[newX][newY]==0){
                        return true;
                    }
                }
        return false;
    }
    public boolean isArea(int[][] matrix,int x,int y){
        return x>=0&&y>=0&&x<matrix.length&&y<matrix[x].length;
    }
}