leetcode-python-day15-第42题

========坚持30天刷leetcode=====

题目链接：https://leetcode-cn.com/problems/trapping-rain-water/

分析：

该题是以数组的形式给出柱子的高度，求柱子的结构能装多少水。

1.首先，明确的是：柱子的数量少于3个将无法装水。

2.对于任一个柱子，使其作为左边界，则必须找到一个大于或等于它的右边界才可以装水。

3.水容量的计算：就约等于一个矩形减去矩形中的小方块。

       借助木桶理论，长（右边界下标-左边界下标-1）；宽（min（左边界，右边界））；小方块(左右边界中所有柱子的高度和)

4. 假如当前柱子是最高的柱子，则其右边所有柱子都小于它，那可咋办？

那就好办啦！只需要在遍历时，记住遇到过最高的柱子和其下标，把它当作是右边界，然后进行水容量计算，搞定！

5.左边界跳到右边界，然后进行新一轮的遍历。水容量累加，最后返回。

代码：

class Solution:
    def trap(self, h):
        llen=len(h)
        if llen<3:
            return 0
        i,j,sum1=0,1,0
        
        while i<llen-1:
            tmpmax,tmpi=-float("inf"),-1
            while j<llen:
                if h[i]>h[j]:
                    if tmpmax<=h[j]:
                        tmpmax,tmpi=h[j],j
                    j+=1
                else:
                    sum1+=(j-i-1)*min(h[i],h[j])-sum(h[i+1:j])
                    i,j=j,j+1
                    break

            if i!=llen-1 and j==llen:
                if tmpi!=-1:
                    sum1+=(tmpi-i-1)*min(h[i],h[tmpi])-sum(h[i+1:tmpi])
                    i,j=tmpi,tmpi+1
        return sum1