417. 太平洋大西洋水流问题(medium) -力扣(leetCode)逆流而上,JS图的深度优先遍历算法

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


⚡️分析⚡️

看到题目矩阵,首先可以想到图的邻接矩阵也是这样,用图的深度优先遍历搜索的思想来解决。
只能从高到低或者在同等高度上流动,即每一个坐标的洋流都可能往上下左右流(如果比它要低,且没超出矩阵范围),因此每个坐标的洋流都可以对他的上下左右继续进行深度优先遍历(如果符合条件)。
那么从哪里开始呢?考虑全面当然是从四条海岸线开始,然后一步步往中间去,途中符合“逆流而上”条件的都为true
“逆流而上”:下一次dfs的高度要比现在高,因为是从四条海岸线往中间遍历,因此中间要高,才能往四条海岸线流。

每一步求解都在代码的注释中,如果听完我分析还不是很懂思路的话,可以去看代码中的注释

代码如下:

/**
 * @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));//填充m*n大小的false数组
    const flow2 = Array.from({ length:m },() => new Array(n).fill(false));

    const dfs = (r,c,flow)=>{
        flow[r][c] = true;
        [[r-1,c], [r+1,c], [r,c-1], [r,c+1]].forEach(([nr,nc]) => {
            if(
                //保证在矩阵中
                nr>=0 && nr<m &&
                nc>=0 && nc<n &&
                //防止死循环,通过数组中的false值来判断有没有遍历过
                !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;
};

算法效率如图:
在这里插入图片描述
时间和空间复杂度都是O(m*n)




觉得该篇文章有用的请不要忘记忘记点击右下角的大拇指~

欢迎大家关注我的公众号:Smooth前端成长记录
公众号正在努力更新CSDN博客内容,想方便阅读博客的C友可以来关注我的公众号以便获得更优良的阅读体验~
在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值