⚡️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友可以来关注我的公众号以便获得更优良的阅读体验~