【数据结构与算法】Leetcode407 接雨水 II -困难(堆、BFS)

接雨水II是对接雨水I的拓展,将一维数组扩展到了二维矩阵,之前接雨水I中的动态规划、单调栈以及双指针在这里都不起作用,但是在我之前介绍接雨水I的文章中对双指针进行了分析和延伸,延伸出的更通用的解法即优先队列解法则可以轻易地移植到接雨水II中。
摘要由CSDN通过智能技术生成

来源:力扣

题目描述

给你一个 m x n 的矩阵,其中的值均为非负整数,代表二维高度图每个单元的高度,请计算图中形状最多能接多少体积的雨水。

示例:
给出如下 3x6 的高度图:
[
  [1,4,3,1,3,2],
  [3,2,1,3,2,4],
  [2,3,3,2,3,1]
]
返回 4

如上图所示,这是下雨前的高度图[[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]] 的状态。

下雨后,雨水将会被存储在这些方块中。总的接雨水量是4

提示:
1 <= m, n <= 110
0 <= heightMap[i][j] <= 20000

优先队列(堆)

  在看这题的解法前可以去我之前讲解接雨水I的那篇博客看一下,那里也应用了这里的解法。
  在那篇博客里我将双指针的本质做了讲解,即可以看作是从数组两端同时注水,然后从外到内,从低到高,直到左右两边的指针相遇即左右两边的水面相遇。
  现在用矩阵取代之前的数组,从一维扩展到了二维,之前的数组两端映射为矩阵的最外围的四边,接雨水I从数组两端注水,那么在这里则是从矩阵最外围的四边同时注水,从外到内,从低到高直至淹没所有位置。
  在接雨水I中可以用双指针指向数组两端,这里则没法为所有外围位置都分配一个指针,所以必须要用到接雨水I从双指针解法延伸出来的优先队列解法。和接雨水I中先将数组两端元素压入队列中类似,先把矩阵最外围四边的所有元素压入队列,然后依次弹出堆顶元素即新的水面值,依次访问上下左右与其相邻的位置,遇到比当前水面值低的就计算积水量,并将新遍历到的元素和新的水面值都压入队列中,重复这些操作直至队列为空,即所有位置都已经遍历过了,用visited数组记录所有遍历过的位置,防止重复计算。这里也用到了BFS广度优先搜索的思想。
  整个过程就像是随着水面不断升高,水不断地从四周向中间漫延一样,最终将所有柱子淹没,完成遍历,此时将水源撤走,留下的水即是矩阵最终的积水量。

代码(Python)

class Solution:
    def trapRainWater(self, heightMap: List[List[
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值