给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
问题解析:
直接按问题描述进行。对于数组中的每个元素,我们找出下雨后水能达到的最高位置,等于两边最大高度的较小值减去当前高度的值。
我们使用动态规划的思想 用数组maxLeft 和maxRight记录左右两遍的最大高度。然后进行遍历 将每个位置接到的雨水累加到res上。
class Solution:
def trap(self, height):
res = 0
l = len(height)
if l<=1:return 0
maxLeft ,maxRight = [0]*l,[0]*l
maxLeft[0],maxRight[l-1] = height[0],height[l-1]
for i in range(1,l):#动态规划记录左边出现的最大值
maxLeft[i] = max(maxLeft[i-1],height[i])
for i in range(l-2,-1,-1):
maxRight[i] = max(maxRight[i+1],height[i])
for i in range(1,l-1): # 累加接到的雨水
res+=max(0,min(maxRight[i],maxLeft[i])-height[i])
return res
时间复杂度O(n),总共遍历了3遍,空间复杂度O(n)