leetcode 778. 水位上升的泳池中游泳

在一个 N x N 的坐标方格 grid 中,每一个方格的值 grid[i][j] 表示在位置 (i,j) 的平台高度。
现在开始下雨了。当时间为 t 时,此时雨水导致水池中任意位置的水位为 t 。你可以从一个平台游向四周相邻的任意一个平台,但是前提是此时水位必须同时淹没这两个平台。假定你可以瞬间移动无限距离,也就是默认在方格内部游动是不耗时的。当然,在你游泳的时候你必须待在坐标方格里面。
你从坐标方格的左上平台 (0,0) 出发。最少耗时多久你才能到达坐标方格的右下平台 (N-1, N-1)?

思路:首先呢,答案啊一定是在[0,N*N-1]内的,看到这里我们就可以对其使用二分(lower_bound)+dfs来做。
还要注意一下两个平台间要水位一样才可以游。

class Solution {
    int[] dx={0,0,1,-1};
    int[] dy={1,-1,0,0};
    int n,m;
    boolean[][] vis;
    boolean flag;

    public int swimInWater(int[][] grid) {
        n=grid.length;
        vis=new boolean[n][n];
        flag=false;
        int l=0;
        int r=n*n-1;
        while (lfor(int i=0;ifalse);
            }
            int mid=l+(r-l)/2;
            dfs(0,0,mid,grid);
            if(flag){
                r=mid;
                flag=false;
            }else{
                l=mid+1;
            }
        }
        return l;
    }

    boolean check(int x,int y){
        if(x<0||y<0||x>=n||y>=n||vis[x][y]){
            return false;
        }
        return true;
    }

    public void dfs(int x,int y,int level,int[][] grid){
        if(flag||grid[x][y]>level){
            return;
        }
        if(x==n-1&&y==n-1){
            flag=true;
            return;
        }
        for(int i=0;i<4;++i){
            int xi=x+dx[i];
            int yi=y+dy[i];
            if(check(xi,yi)&&grid[xi][yi]<=level) {
                vis[x][y]=true; 
                dfs(x+dx[i],y+dy[i],level,grid);
            }
        }
    }
}

你可能感兴趣的:(DFS)