代码思路:
1.使用动态规划,需要对每个下标位置都向两边扫描。如果已经知道每个位置两边的最大高度,则可以在 O(n)的时间内得到能接的雨水总量。空间复杂度是O(n),其中n是数组长度,空间复杂度由栈空间决定。
2.可以使用双指针和两个变量代替两个数组,这样空间复杂度降为O(1)。
注意到下标 i 处能接的雨水量由leftMax[i]和rightMax[i]中的最小值决定。
class Solution:
def trap(self, height: List[int]) -> int:
ans=0
left=0
right = len(height)-1
pre_max = 0
suf_max = 0
while left < right:
pre_max = max(pre_max,height[left])
suf_max = max(suf_max,height[right])
if pre_max < suf_max:
ans += pre_max-height[left]
left += 1
else:
ans += suf_max - height[right]
right -= 1
return ans