leecode 力扣刷题 #42 接雨水 (困难)

#42：

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/trapping-rain-water/

这题属于困难，参考了解题区一位数学大佬的解题思路：https://leetcode-cn.com/problems/trapping-rain-water/solution/wei-en-tu-jie-fa-zui-jian-dan-yi-dong-10xing-jie-j/，本来很困不想写了，看完这个思路以后就觉得简单很多，果然人脑和人脑是不一样的。整理一下，大概如下：

解题思路：

1. 简单举个例子：height=[2,1,0,3,1]，如下图：

2. 下雨方向从左往右，依次填满，S1=墙的面积+雨水的面积=2+2+2+3+3 =12
下雨方向从右往左，依次填满，S2=墙的面积+雨水的面积=1+3+3+3+3 =13


3. 这部分可以根据**韦恩图(venn)**来理解，如下图，S1+S2 包含了整个矩形，而且重合部分正好是墙和水的面积加总，所以：S1 + S2 = S墙 + S水 + S矩形，所以：S水 = S1 + S2 - S矩形 - S墙

4. S矩形 = list的长度 × list内最大值 = 5×3 = 15，S墙 = list内数字的加总 = sum[2,1,0,3,1] = 7，S水 = 12 + 13 - 15 - 7 = 3，和上面蓝色的3格相符，应该是正确的，最后写code测测看是不是这样就可以了。

实测：

class Solution(object):
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        #方向从左往右
        s1=0 #初始面积设为0
        l1=0 #方向从左往右，墙的最大高度，初始设为0
        for i in range(len(height)):#i分别是第0,1，2,3，4根墙柱子
            if height[i]>l1:#如果第i根墙柱子大于目前最大高度l1，
                l1=height[i]#则最大高度变为第i根墙柱子
                #s1+=l1
            s1+=l1#加总
        #print(s1)
        
		#方向从右往左，原理和上面一样
        height2=height[::-1]#list倒序排列, 模拟从右往左的墙，height=[2,1,0,3,1]，所以height2=[1,3,0,1,2]
        s2=0
        l2=0
        for j in range(len(height2)):
            if height2[j]>l2:
                l2=height2[j]
                #s2+=l2
            s2+=l2
        #print(s2)

        #print(s1+s2-len(height)*max(height)-sum(height))
        return s1+s2-len(height)*max(height)-sum(height)

leecode实测截图：