太平洋大西洋水流问题有一个 m × n 的矩形岛屿，与太平洋和大西洋相邻。 “太平洋” 处于大陆的左边界和上边界

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第24天，点击查看活动详情

有一个 m × n 的矩形岛屿，与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界，而 “大西洋” 处于大陆的右边界和下边界。

这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights ， heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。

岛上雨水较多，如果相邻单元格的高度 小于或等于 当前单元格的高度，雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。

返回网格坐标 result 的 2D列表 ，其中 result[i] = [ri, ci] 表示雨水可以从单元格 (ri, ci) 流向 太平洋和大西洋 。

示例 1：

输入: heights = [[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]]

示例 2：

输入: heights = [[2,1],[1,2]]
输出: [[0,0],[0,1],[1,0],[1,1]]

提示：

m == heights.length
n == heights[r].length
1 <= m, n <= 200
0 <= heights[r][c] <= 10^5

思路

本题可以使用深度优先遍历求解。定义两个二维数组pacifc和atlantic分别存储小岛[r,c]是否可以流入太平洋和大西洋，初始值填充false。先遍历可以流入太平洋的小岛，有题目可知第0排和第0列可以直接流入太平洋，即pacifc[0][c] = pacifc[r][0] = true。我们以这些坐标作为起点，遍历它上下左右四个方向，如果高度大于等于当前坐标点的高度，说明遍历的小岛也可以流入太平洋，赋值true，再以这个小岛为起点，进行同样的遍历。最后pacifc中为true的坐标点就是可以流入太平洋的小岛坐标点。用同样方法从下侧和右侧开始遍历可以流入大西洋的小岛。最后pacifc和atlantic中同时为true的坐标点就是我们要求的坐标点。

解题

/**
 * @param {number[][]} heights
 * @return {number[][]}
 */
var pacificAtlantic = function (heights) {
  const m = heights.length;
  const n = heights[0].length;
  const pacifc = new Array(m).fill(0).map(() => new Array(n).fill(false));
  const atlantic = new Array(m).fill(0).map(() => new Array(n).fill(false));
  const res = [];
  const dirs = [
    [-1, 0],
    [1, 0],
    [0, 1],
    [0, -1],
  ];
  const dfs = (row, col, ocean) => {
    if (ocean[row][col]) return;
    ocean[row][col] = true;
    const h = heights[row][col];
    dirs.forEach(([i, j]) => {
      const x = row + i;
      const y = col + j;
      if (x >= 0 && x < m && y >= 0 && y < n && heights[x][y] >= h) {
        dfs(x, y, ocean);
      }
    });
  };
  for (let i = 0; i < n; i++) {
    dfs(0, i, pacifc);
    dfs(m - 1, i, atlantic);
  }
  for (let i = 0; i < m; i++) {
    dfs(i, 0, pacifc);
    dfs(i, n - 1, atlantic);
  }
  
  for (let i = 0; i < m; i++) {
    for (let j = 0; j < n; j++) {
      if (pacifc[i][j] && atlantic[i][j]) {
        res.push([i, j]);
      }
    }
  }
  return res;
};