-
昨天睡前 刷 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)