Leetcode刷题(第417题)——太平洋大西洋水流问题

x33g5p2x  于2022-03-31 转载在 其他  
字(0.9k)|赞(0)|评价(0)|浏览(236)

一、题目

二、示例

  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]]
  2. 输出: [[0,4],[1,3],[1,4],[2,2],[3,0],[3,1],[4,0]]
  1. 输入: heights = [[2,1],[1,2]]
  2. 输出: [[0,0],[0,1],[1,0],[1,1]]

三、思路
1、这题采用图的深度优先遍历
2、创建两个二维矩阵,其中的值都设置为false
3、分别从大西洋和太平洋海岸线起步,经过之处节点的值设置为true
4、最后再将两个图进行比对,拿出两张图的节点都是true的节点并保存
5、最后返回
四、代码

  1. /**
  2. * @param {number[][]} heights
  3. * @return {number[][]}
  4. */
  5. var pacificAtlantic = function(heights) {
  6. let row = heights.length
  7. let col = heights[0].length
  8. let flow1 = new Array(row).fill(0).map(item => new Array(col).fill(false))
  9. let flow2 = new Array(row).fill(0).map(item => new Array(col).fill(false))
  10. const rec = (x, y, flow) => {
  11. if(x < 0 || x >= row || y < 0 || y >= col) return
  12. flow[x][y] = true
  13. ;[[x, y + 1],[x + 1, y], [x, y - 1], [x -1, y]].forEach(([xp, yp]) => {
  14. if(heights[xp][yp] >= heights[x][y]) {
  15. rec(x, y, flow)
  16. }
  17. })
  18. }
  19. for(let i = 0; i < row; i++) {
  20. rec(i, 0, flow1)
  21. rec(i, col - 1, flow2)
  22. }
  23. for(let j = 0; j < col; j++) {
  24. rec(0, j, flow1)
  25. rec(row - 1, j, flow2)
  26. }
  27. console.log(flow1)
  28. console.log(flow2)
  29. };

五、总结

相关文章