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

Posted 2022-01-14 tu9oh0st

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]);