时间复杂度:O(n);空间复杂度:O(1)
解题思路
初见题,就令我想起了“0011.接最多水的容器”这道题,该题题解可参考我之前写的LeetCode0011.盛最多水的容器Go语言AC笔记https://blog.csdn.net/Hexa_H/article/details/126166067 所以受该题影响,我还是想尝试用双指针解决现在这一题目,不过在如何计算雨水上又犯了难。但看过题解后醍醐灌顶,非常顺利地写出了AC代码。
这一题双指针的移动还是k和11题一样,左右指针比较,谁的高度小谁就向中间移动,不过这一题在移动指针前需要做一下判断,需要确定当前高度是否是左/右指针遍历过的最高高度。下面以左指针高度小于右指针高度举例,右指针高度小于左指针高度同理。如果是前面说的情况,当前左指针高度是左指针遍历过的最高左指针高度,那么就不用接雨水了,因为该情况也接不到雨水(左侧高度都比当前高度低,也接不住雨水呀,雨水都顺着斜坡流走了),但如果当前高度不是左指针遍历过的最高高度,这就说明左侧有一堵墙比现在的高度高(并且右侧有一堵墙比左侧最高的墙还要高,具体原因可以先自己想想,在之后会细说),雨水就可以积攒在当前左指针的墙上,但是水加墙的高度不能超过左侧最高墙的高度,不然水会溢出,所以我们就可以用maxLeft-height[left]得到左指针积攒的雨水体积。
上面的这个图可以很好地帮助理解解题思路,左指针不断向中间移动,因为它的高度一直小于右指针高度,在移动时不能计算接到的雨水体积,只有当left==3时,左指针的高度终于不是左指针遍历过的最高高度了,所以可以用左指针遍历过的最高高度(即left==2时的高度)减去当前左指针高度得到该列接住的雨水体积。
一定会有小伙伴问为什么接住的雨水体积要用左侧最高高度减当前左指针高度,如果此时左侧最高墙右侧的所有墙高度都低于这个墙的高度,那么不就不能保证当前左指针接住的雨水能和左侧最高墙的高度齐平吗?其实这个问题在之前就已经回答过了,只有在左指针高度小于右指针高度的情况下,我们才会移动左指针并且在这之前计算左指针最高高度以及接住的雨水体积,所以右指针的高度一定是高于左侧最高墙高度的。
当左右指针碰面时,遍历结束,返回计算好的雨水体积。
AC代码
func trap(height []int) int {
left,right,maxLeft,maxRight,ans:=0,len(height)-1,0,0,0
for left<right{
if height[left]<height[right]{
if height[left]>maxLeft{
maxLeft=height[left]
}else{
ans+=maxLeft-height[left]//当前该列接住的雨水体积就是当前的左指针最高高度与当前左指针高度差
}
left++
}else{
if height[right]>maxRight{
maxRight=height[right]
}else{
ans+=maxRight-height[right]//当前该列接住的雨水体积就是当前的右指针最高高度与当前右指针高度差
}
right--
}
}
return ans
}
感悟
比较左右指针高度,矮的那个指针向对方移动,同时记录遍历过的最高值,最高值与当前值的差值就是能接的雨水高度。
又一次被双指针折服,能想到双指针的移动却没有想到如何去计算接住的雨水体积,差一步都不行啊。