Leetcode 42 接雨水

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

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

解题思路1 ：找到最大值，然后分别向两边计算雨量。

class Solution:
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        res, hei_len, peak = 0, len(height), 0
        if hei_len < 3:
            return 0
        max_val = height[0]
        for i in range(1, hei_len):
            if height[i] >= max_val:
                max_val = height[i]
                peak = i
        
        max_left, max_right = height[0], height[-1]
        for i in range(peak):
            if max_left < height[i]:
                max_left = height[i]
            else:
                res += max_left - height[i]
                
        for i in range(hei_len-1, peak, -1):
            if max_right < height[i]:
                max_right = height[i]
            else:
                res += max_right - height[i]
            
        return res

解题思路2：从开头和结尾 同时向中间找，双指针法。

class Solution:
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        left, right = 0, len(height) - 1
        left_max, right_max = 0, 0
        res = 0
        while left < right:
            if height[left] < height[right]:
                if height[left] >= left_max:
                    left_max = height[left]
                else:
                    res += left_max - height[left]
                left += 1
            else:
                if height[right] >= right_max:
                    right_max = height[right]
                else:
                    res += right_max - height[right]
                right -= 1
                
        return res