JavaScript算法31- 太平洋大西洋水流问题（leetCode:417中等）

Y_soybean_milk

已于 2022-05-23 18:03:10 修改

阅读量1.5k

点赞数 1

分类专栏：每日算法文章标签：算法 leetcode 深度优先图

于 2022-05-23 18:00:48 首次发布

本文链接：https://blog.csdn.net/y_soybean_milk/article/details/124931833

版权

每日算法专栏收录该内容

45 篇文章 7 订阅

订阅专栏

该博客介绍了一种解决417.太平洋大西洋水流问题的方法，通过深度优先搜索（DFS）策略，从矩阵的海岸线开始逆流而上，标记可以到达太平洋和大西洋的单元格。算法首先初始化两个矩阵记录流向，然后沿着四个海岸线进行DFS遍历，最后收集同时可以流向两个大洋的坐标。时间复杂度为O(m*n)，空间复杂度同样为O(m*n)。

摘要由CSDN通过智能技术生成

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

一、题目

有一个 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⁵

二、思路

把矩阵想象成图
从海岸线逆流而上遍历图，所到之处就是可以流到某个大洋的坐标

具体步骤

新建两个矩阵，分别记录能流到两个大洋的坐标
从海岸线，多管齐下，同时深度优先遍历图，过程中填充上述矩阵
遍历两个矩阵，找出能流到两个大洋的坐标

三、题解

/**
 * @param {number[][]} heights
 * @return {number[][]}
 */
var pacificAtlantic = function (heights) {
    if (!heights || !heights[0]) { return []; }
    const m = heights.length
    const n = heights[0].length
    const flow1 = Array.from({ length: m }, () => new Array(n).fill(false))
    const flow2 = Array.from({ length: m }, () => new Array(n).fill(false))

    // r:row, c:column
    const dfs = (r, c, flow) => {
        flow[r][c] = true
        const nextArr = [[r - 1, c], [r + 1, c], [r, c - 1], [r, c + 1]]
        nextArr.forEach(([nr, nc]) => {
            if (
                // 保证在矩阵中
                nr >= 0 && nr < m &&
                nc >= 0 && nc < n &&
                // 防止死循环
                !flow[nr][nc] &&
                // 保证逆流而上
                heights[nr][nc] >= heights[r][c]
            ) {
                dfs(nr, nc, flow)
            }
        })
    }

    // 沿着四条海岸线逆流而上
    for (let r = 0; r < m; r++) {
        dfs(r, 0, flow1)
        dfs(r, n - 1, flow2)
    }
    for (let c = 0; c < n; c++) {
        dfs(0, c, flow1)
        dfs(m - 1, c, flow2)
    }

    // 收集能流到两个大洋里的坐标
    const res = []
    for (let r = 0; r < m; r++) {
        for (let c = 0; c < n; c++) {
            if (flow1[r][c] && flow2[r][c]) {
                res.push([r, c])
            }
        }
    }
    return res;
};