(day23)leecode热题——接雨水

描述

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

示例 1：

输入：height = [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 个单位的雨水（蓝色部分表示雨水）。 

示例 2：

输入：height = [4,2,0,3,2,5]
输出：9

提示：

• n == height.length
• 1 <= n <= 2 * 104
• 0 <= height[i] <= 105

碎碎念：此题是盛最多水的容器一题的升级版，存在左右边界的都可作为容器来接水。雨水的体积即为形成容器的体积 -- 容器内部的所有heiht[i]，跟俄罗斯方块有一拼哈哈

前后缀分解

class Solution:
    def trap(self, height: List[int]) -> int:
        
        # 获取高度数组的长度
        n = len(height)
        
        # 初始化 pre_max 数组，存储每个位置左边的最大高度
        pre_max = [0] * n
        # 第一个位置的左边最大高度就是它本身
        pre_max[0] = height[0]
        # 遍历数组，从左向右更新 pre_max 数组
        for i in range(1, n):
            pre_max[i] = max(pre_max[i - 1], height[i])
        
        # 初始化 suf_max 数组，存储每个位置右边的最大高度
        suf_max = [0] * n
        # 最后一个位置的右边最大高度就是它本身
        suf_max[-1] = height[-1]
        # 遍历数组&