接雨水
1.思路
盛最多的水的容易问题升级版!
这个问题可以这样模拟:我们每一格能盛放的水的容量其实取决于这一格的左边的单位高度和右边的单位高度,拿第三个格举例,左边的单位高度为1,右边的单位高度为2,同时这一格是没有黑色柱子的,所以这一格能存储的容量就是min(左边高度—右边高度)— 黑色柱子的高度!而左边的高度就是当前位置左边的单位高度的最大值,同理,右边的高度也相当于当前位置右边的单位高度的最大值。为什么要取左右两边高度的最小值呢?因为取了最大值,多余的水也会溢出,所以取最大值毫无意义。有了这种想法,我们就可以动手写出解决该问题的代码!
2.解题方法
两个循环遍历记录下当前位置左边的单位高度的最大值pre_max以及当前位置右边的单位高度最大值suf_max,问题的解就等于每一格的min(pre_max,suf_max)减去当前格子的黑色柱子的高度。
3.复杂度
时间复杂度:O(n)
空间复杂度:O(n)
4.Code
class Solution:
def trap(self, height: list[int]) -> int:
len_height=len(height)
pre_max=[0]*len_height
pre_max[0]=height[0]
suf_max=[0]*len_height
suf_max[-1]=height[-1]
for i in range(1,len_height):
pre_max[i]=max(height[i],pre_max[i-1])
for j in range(len_height-2,-1,-1):
suf_max[j]=max(height[j],suf_max[j+1])
ans=0
for x,y,z in zip(pre_max,suf_max,height):
ans+=min(x,y)-z
return ans