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

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]] (上图中带括号的单元).

分析

左边和上边是太平洋，右边和下边是大西洋，内部的数字代表海拔，海拔高的地方的水能够流到低的地方，求解水能够流到太平洋和大西洋的所有位置。

贴出代码

class Solution {
    int m ,n;
    private int[][] matrix;
    private int[][] direction = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};

    public List<int[]> pacificAtlantic(int[][] matrix) {
        List<int[]> 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(matrix,i, 0, canReachP);
//            dfs(matrix,i, n - 1, canReachA);
//        }
//
//        for (int i = 0; i < n; i ++){
//            dfs(matrix,0,i,canReachP);
//            dfs(matrix,m - 1, i, canReachA);
//        }

        for (int i = 0; i < n; i ++){
            for (int j = 0; j < m; j ++){
                if (i ==0 || j == 0){
                    dfs(matrix,i,j,canReachP,Integer.MIN_VALUE);
                }
                if (i == n - 1 || j == m -1){
                    dfs(matrix,i,j,canReachA,Integer.MIN_VALUE);
                }
            }
        }

        for (int i = 0; i < m; i ++){
            for (int j = 0;j < n; j ++){
                if (canReachP[i][j] && canReachA[i][j]){
                    ret.add(new int[]{i,j});
                }
            }
        }
        return ret;
    }

    private void dfs(int[][] matrix, int r,int c, boolean[][] canReach,int height){
        if (r < 0 || c < 0 ||r >= m || c >= n || matrix[r][c] > height || canReach[r][c]){
            return;
        }

        canReach[r][c] = true;
        for (int[] d : direction){
            dfs(matrix,r + d[0],c + d[1],canReach,matrix[r][c]);
        }
    }
}

转载于:https://www.cnblogs.com/Tu9oh0st/p/11029484.html