leetcode 42 接雨水 python

每天坚持刷题！！

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

题目分析
1. 首先明确如何计算一个可以接雨水的结构的面积，第一需要两个边界，类似桶的两个边，注意此时取得是两个边界的最小值为最终边界的高，然后用边界的高乘以两个边界的距离再减去两个边界之间所有柱子的面积即可。
2. 从上面的分析我们需要在遍历时寻找两个边界，同时记录两个边界的位置。两个边界的高可以分为三种情况，第一是两者相同，第二是左高右低，第三是左低右高，所有可以接雨水的局部结构都具有以上三种情况的一种，那么针对第二种情况可以从数组的末尾处倒序遍历，首先存储目前最大的边界和位置，然后向前遍历直到找到一个值比现在最大的边界还要大，那么此时存储的最大边界和现在找到的边界就构成了一个接雨水的局部结构，按照1的方法计算雨水的面积，然后继续遍历。针对第三种的情况类似，只不过变成从数组的开头到末尾的方向开始寻找。针对第一种情况可以在从头到尾遍历或者从尾到头遍历中把大于号变成大于等于即可。

class Solution(object):
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        if not height:  # 边界检查
            return 0
        result = 0
        temp_max = None  # 首先从左到右遍历
        for id, i in enumerate(height):
            if temp_max is None:
                temp_max = [id, i]
            else:
                if i > temp_max[1]:
                    if id - temp_max[0] != 1:
                        result += (id - temp_max[0] - 1) * temp_max[1] - sum(height[temp_max[0] + 1: id])
                    temp_max = [id, i]
        temp_max = None # 然后从右到左遍历
        for j in xrange(len(height) - 1, -1, -1):
            if temp_max is None:
                temp_max = [j, height[j]]
            else:
                if height[j] >= temp_max[1]:
                    if temp_max[0] - j != 1:
                        result += (temp_max[0] - 1 - j) * temp_max[1] - sum(height[j + 1: temp_max[0]])
                    temp_max = [j, height[j]]
        return result