力扣刷题：42.接雨水

题目：

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

示例 1：

输入：height = [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 个单位的雨水（蓝色部分表示雨水）。 
示例 2：

输入：height = [4,2,0,3,2,5]
输出：9
 

提示：

n == height.length
1 <= n <= 2 * 104
0 <= height[i] <= 105

分析：

参考力扣的解法，我把题目理解成，先建立三个列表：height 用于记录每个位置的柱高；leftheight用于记录每个位置左边的最高柱；rightheight用于记录每个位置右边的最高柱。由常识可以知道，储水量由两边的最短柱减去自身的高度就可以得到了。所以，本题只需要求出两边的最高柱即可。

核心数学公式：min(leftheight[i],rightheight[i])-height[i]

代码：

import numpy as np
import pandas as pd
import math
#coding='utf-8'
a = [0,1,0,2,1,0,1,3,2,1,2,1]
class Solution(object):
    def __init__(self,height):
        self.height = height
    def trap(self):
        leftMax = [0]
        rightMax = []
        sum = 0
        n = len(self.height)
        for i in range(1,n):
            leftMax.append(max(leftMax[i-1],self.height[i]))

        for i in range(1,n):
            rightMax.append(0)
        rightMax.append(1)
        for i in range(n-2,-1,-1):
            rightMax[i] = max(rightMax[i+1],self.height[i])
            print(rightMax)

        print("leftMax:",leftMax)
        print("rightMax:",rightMax)

        for i in range(n):
            sum = sum+min(rightMax[i], leftMax[i]) - self.height[i]
        print(sum)

solution1 = Solution(height=a)  #每个位置对应的柱子高度
solution1.trap()