leetcode 42. 接雨水

题目描述

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

上面是由数组 [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

 

解题思路

主要难点是想明白雨水的累积过程。

1、在某个位置i处，它能积累的最大雨水取决于左右两边最大高度中的最小值

2、左边的最大值小于右边的最大值时，当前位置i所能积累的雨水量为左边的最大值剪去当前高度；判读左边高度的下一个位置

3、左边的最大值大于等于右边的最大值时，当前位置i所能积累的雨水量为右边的最大值剪去当前高度；判断右边高度的前一个位置

## 为什么要剪去当前高度，因为当前高度存在时，雨水的积累量是起决定作用
## 的左边或右边高度最大值剪去当前高度。
## 有多少当前高度，就要少积累多少雨水。
class Solution:
    def trap(self, height: List[int]) -> int:
        left = 0
        right = len(height) - 1
        ans = 0

        leftmax = rightmax = 0
        while left <= right:
            if leftmax < rightmax:
                ans += max(0,leftmax-height[left])
                leftmax = max(leftmax,height[left])
                left += 1
            else:
                ans += max(0,rightmax-height[right])
                rightmax = max(rightmax,height[right])
                right -= 1

        return ans