leetcode刷题记录 2018.7.6

1.接雨水问题

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

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 感谢 Marcos 贡献此图。

示例:

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

分析：根据题目中的要求，首先想到的是，要能存住水，一定要满足两边的比中间高。所以我们可以先找出最高的，然后从两边向最高的靠近，定义一个变量记录两边的次高值，当次高值的下一个值比次高值小，肯定可以存住水，如果发现了更高的值，就更新次高值，直到逼近最高值结束。由此可以写程序

class Solution:
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        second = 0
        result = 0
        maxid = 0
        for i in range(1, len(height)):
            if height[i] > height[maxid]:
                maxid = i
        for i in range(0, maxid):
            if second < height[i]:
                second = height[i]
            else:
                result += second - height[i]
        second = 0
 
        for j in range(len(height) - 1, maxid, -1):
            if second < height[j]:
                second = height[j]
            else:
                result += second - height[j]
        return result