[leetCode]417. 太平洋大西洋水流问题

题目

链接：https://leetcode-cn.com/problems/pacific-atlantic-water-flow

给定一个 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

思路:
  这题与130. 被围绕的区域大同小异, 以边界为起始点进行深度优先搜索，使用辅助数组记录能到达的坐标，最后通过遍历两个辅助数组来判断符合要求的坐标点。

class Solution {
    int rows;
    int cols;
    int[][] dirs = new int[][]{{-1, 0},{0, -1},{1, 0},{0, 1}};
    List<List<Integer>> result =  new ArrayList<>();
    public List<List<Integer>> pacificAtlantic(int[][] matrix) {
        if (matrix.length == 0) return result;
        rows = matrix.length;
        cols = matrix[0].length;

        boolean[][] canReachP = new boolean[rows][cols];
        boolean[][] canReachA = new boolean[rows][cols];

        for (int i = 0; i < rows; i++) {
            dfs(matrix, i, 0, canReachP);
            dfs(matrix, i, cols - 1, canReachA);
        }

        for (int i = 0; i < cols; i ++) {
            dfs(matrix, 0, i, canReachP);
            dfs(matrix, rows - 1, i, canReachA);
        }

        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < cols; j++) {
                if (canReachA[i][j] && canReachP[i][j]) {
                    result.add(Arrays.asList(i,j));
                }
            }
        }
        return result;
        
    }

    private void dfs(int[][] matrix, int x, int y, boolean[][] reach) {
        if (reach[x][y]) {
            return;
        }
        reach[x][y] = true;
        for (int[] dir : dirs) {
            int nx = x + dir[0];
            int ny = y + dir[1];
            if (nx < 0 || nx >= rows || ny < 0 || ny >= cols
            || matrix[x][y] > matrix[nx][ny]) {
                continue;
            }
            dfs(matrix, nx, ny, reach);
        }
    }
}