LeetCode--Trapping Rain Water（捕获雨水）Python

题目：

假设n个非负整数代表一个海拔图，每个条的宽度为1，计算它能在雨后捕获多少水。

For example,

例如,

lì例 rú如 ,

Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.

给定数组[ 0,1,0,2,1,0,1,3,2,1,2,1 ]，海拔图如下，返回6。

解题思路：

观察该海拔图可以发现被水填满后的形状是先升后降的塔形，因此，先遍历一遍找到塔顶，然后分别从两边开始，往塔顶所在位置遍历，水位只会增高不会减小，且一直和最近遇到的最大高度持平，这样就可以知道当前遍历到的位置的水位为多少，故而再遍历一遍求和即可得到最终的储水量。

代码（Python）：

class Solution(object):
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        max_h = 0
        index_max = 0
        n = len(height)
        
        for i in range(n):
            if height[i]>max_h:
                max_h = height[i]
                index_max = i
        
        temp_h = 0
        output = 0
        for i in range(index_max):
            if temp_h<height[i]:
                temp_h = height[i]
                continue
            output = output+temp_h-height[i]
            
        temp_h = 0        
        for i in range(n-index_max-1):
            if temp_h<height[-i-1]:
                temp_h = height[-i-1]
                continue
            output = output+temp_h-height[-i-1]
            
        return output