042.trapping-rain-water

给 n 个非负整数表示每个宽度为一的柱子的高度图,计算下雨之后能接多少水。

例如,
输入 [0,1,0,2,1,0,1,3,2,1,2,1],返回 6

上面的高度图由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示,在这种情况下,可以接 6 个单位的雨水(蓝色部分)。 


解题思路:

    这道题目设计的非常巧妙,主要需要解决的问题是如何让左右两根很高的柱子中间填满,可以从左右同时网中间遍历,同时要找到左右两边最高的那根柱子,因为中间的都是要填满的。


解题代码:

        leftwall=rightwall=float("-inf")
        left=0
        right=len(height)-1
        ans =0
        while left<=right:
            if leftwall<=rightwall:
                ans+=max(0,leftwall-height[left])
                leftwall=max(leftwall,height[left])
                left+=1
            else:
                ans+=max(0,rightwall-height[right])
                rightwall=max(rightwall,height[right])
                right-=1
        return ans

你可能感兴趣的:(Leetcode)