leetcode42 接雨水（Python实现）

题目：给定 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/trapping-rain-water

黑色为给定的柱子，蓝色的地方为可以接的雨水。
代码实现：

from typing import List

"""
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
"""
# 数学思想：观察接完雨水后的柱子图，可与发现柱子图为最大值左边非单调递增，右边为非单调递减
class Solution:
    def trap(self, height: List[int]) -> int:
        if len(height) == 0: return 0
        max_element = height.index(max(height))
        curr_left, curr_right, res = 0, 0, 0
        for i in range(max_element):
            if height[i] > curr_left:
                curr_left = height[i]
            else:
                res += curr_left - height[i]

        for j in range(len(height) - 1, max_element, -1):
            if height[j] > curr_right:
                curr_right = height[j]
            else:
                res += curr_right - height[j]
        return res

实现思路：

1. 找到数组最大值坐标
2. 分别从数组最左边和最右边遍历到最大值处。
3. 构造单调非递增（递减）序列
4. 补出的值即为接到的雨水量
   思路来自评论区断，舍，离大佬，清晰明了，受教