leetcode 407. 接雨水 II【BFS】

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

示例 1:

输入: heightMap = [[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]]
输出: 4
解释: 下雨后,雨水将会被上图蓝色的方块中。总的接雨水量为1+2+1=4。

示例 2:

输入: heightMap = [[3,3,3,3,3],[3,2,2,2,3],[3,2,1,2,3],[3,2,2,2,3],[3,3,3,3,3]]
输出: 10

提示:

m == heightMap.length
n == heightMap[i].length
1 <= m, n <= 200
0 <= heightMap[i][j] <= 2 * 10^4

思路:
maxHeight记为矩阵中最高的柱子高度,
water[i][j]记为每根柱子的蓄水后的高度,
qu[(i, j)]记为邻居待更新的柱子的队列,
dir用于确定四周邻居的方位。

初始化:每根柱子的蓄水后高度初始化为maxHeight,边缘的柱子蓄水后的高度实际等于自身原始的高度,即蓄水量为0.

BFS:更新队列中柱子的邻居的蓄水后高度.每个邻居的water[i][j]=max(heightMap[i][j], min(water[i-1][j]), water[i+1][j], water[i][j-1], water[i][j+1]),直到队列中不再有带更新柱子。

code:

class Solution:
    def trapRainWater(self, heightMap: List[List[int]]) -> int:
        m, n = len(heightMap), len(heightMap[0])
        maxHeight = max(max(row) for row in heightMap)  # 矩阵中最高的柱子.
        water = [[maxHeight for _ in range(n)] for _ in range(m)]  # 每根柱子的蓄水后的高度初始化为maxHeight.
        qu = []  # 邻居待更新的柱子的队列.
        dir = [-1, 0, 1, 0, -1]  # 确定四周邻居的方位.

        # 边缘的柱子蓄水后的高度实际等于自身原始的高度,即蓄水量为0.
        for i in range(m):
            for j in range(n):
                if i == 0 or i == (m-1) or j == 0 or j == (n-1):
                    water[i][j] = min(maxHeight, heightMap[i][j])
                    qu.append((i, j))

        # BFS. 更新队列中柱子的邻居的蓄水后高度.
        # 每个邻居的water[i][j]=max(heightMap[i][j], min(water[i-1][j]), water[i+1][j], water[i][j-1], water[i][j+1]).
        # 直到队列中不再有带更新柱子
        while len(qu) > 0:
            (x, y) = qu.pop(0)
            for k in range(4):
                nx, ny = x + dir[k], y + dir[k+1]
                if nx == -1 or ny == -1 or nx == m or ny == n:
                    continue
                if water[nx][ny] > water[x][y] and water[nx][ny] > heightMap[nx][ny]:
                    water[nx][ny] = max(heightMap[nx][ny], water[x][y])
                    qu.append((nx, ny))
        ans = 0
        for i in range(m):
            for j in range(n):
                ans += water[i][j] - heightMap[i][j]
        return ans
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值