接雨水 LeetCode 算法图解

• 昨天睡前 刷 LeetCode 题，看到个大佬写的，接雨水的答案很简洁，所以今天有空又去看了下，顺手做了个图解。

• 问题: 给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图。

在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）

• 这是大佬写的答案

# from 狂奔的蜗牛
def trap(self, height: List[int]) -> int:
    ans = 0
    h1 = 0
    h2 = 0
    for i in range(len(height)):
        h1 = max(h1,height[i])
        h2 = max(h2,height[-i-1])
        ans = ans + h1 + h2 -height[i]
    return  ans - len(height)*h1

• 我们来画个图来解释一下。

• 首先灌满右半部分水
  

• 再灌左半部分水

• 叠加起来

• 可以看出 黑色与黄色面积被覆盖了两次

• 移除掉 最上面全部覆盖的一层，

• 再去掉整个数组

• 最终获得 我们需要的答案

• 再把大佬的函数写简洁一点

def trap(self, height: List[int]) -> int:
    lh, rh, layer = 0,0,0 # 初始化
    for i in range(len(height)):
        lh = max(lh,height[i]) # 左最高值
        rh = max(rh,height[-i-1]) # 右最高值
        layer += lh + rh
    return  layer - len(height)*lh -sum(height)