leetcode 417. 太平洋大西洋水流问题【DFS】

leetcode 417. 太平洋大西洋水流问题

给定一个 m x n 的非负整数矩阵来表示一片大陆上各个单元格的高度。“太平洋”处于大陆的左边界和上边界,而“大西洋”处于大陆的右边界和下边界。

规定水流只能按照上、下、左、右四个方向流动,且只能从高到低或者在同等高度上流动。
请找出那些水流既可以流动到“太平洋”,又能流动到“大西洋”的陆地单元的坐标。

提示:

输出坐标的顺序不重要
m 和 n 都小于150

示例:

给定下面的 5x5 矩阵:

  太平洋 ~   ~   ~   ~   ~ 
       ~  1   2   2   3  (5) *
       ~  3   2   3  (4) (4) *
       ~  2   4  (5)  3   1  *
       ~ (6) (7)  1   4   5  *
       ~ (5)  1   1   2   4  *
          *   *   *   *   * 大西洋

返回:

[[0, 4], [1, 3], [1, 4], [2, 2], [3, 0], [3, 1], [4, 0]] (上图中带括号的单元).

思路:

乍一看,不像是常见的dfs题型,所以我们要做的就是把它转换为我们熟悉的思路。
要找到这些满足条件的点,不一定得全部遍历dfs一遍验证该点是否都满足条件,其实可以反过来想想。
从四周毗邻海洋的点开始,看从这些点开始能dfs到哪些点(也就是这里的水可以从哪里流过来),设置相应的canReach二维数组。这样就变成常见的dfs了。

private int m, n;
    private int[][] matrix;
    private int[][] direction = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};
    public List<List<Integer>> pacificAtlantic(int[][] matrix) {
        List<List<Integer>> ret = new ArrayList<>();
        if(matrix==null || matrix.length == 0)
            return ret;
        m = matrix.length;
        n = matrix[0].length;
        this.matrix = matrix;
        boolean[][] canReachP = new boolean[m][n];
        boolean[][] canReachA = new boolean[m][n];

        for(int i = 0; i<m;i++){
            dfs(i, 0, canReachP);
            dfs(i, n-1, canReachA);
        }
        for(int i = 0;i<n;i++){
            dfs(0, i, canReachP);
            dfs( m-1, i, canReachA);
        }
        for (int i = 0;i<m;i++){
            for(int j = 0;j<n;j++){
                if (canReachP[i][j] && canReachA[i][j])
                    ret.add(Arrays.asList(i,j));
            }
        }
        return ret;
    }
    private void dfs(int r, int c, boolean[][] canReach){
        if (canReach[r][c])
            return;
        canReach[r][c]=true;
        for(int[] d : direction){
            int nextR = d[0]+r;
            int nextC = d[1]+c;
            if(nextR<0 || nextR>=m || nextC<0 || nextC>=n || matrix[r][c]>matrix[nextR][nextC])
                continue;
            dfs(nextR,nextC,canReach);
        }
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值