42. 接雨水(困难)

https://leetcode.cn/problems/trapping-rain-water/

给定 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

class Solution:
    def trap(self, height: List[int]) -> int:
        # 时间复杂度：O(n)
        # 空间复杂度：O(n)
        # 难度：☆☆☆☆☆
        # 标签：前后缀分解
        # 思路：遍历每个柱子h，柱子前最大值和柱子后最大值，取最小min(pre_max, suf_max)
        # min(pre_max, suf_max) - h，就是这个柱子的位置能装多少水
        # n = len(height)
        # ret = 0

        # pre_max = [0] * n  # pre_max[i] 表示从 height[0] 到 height[i] 的最大值
        # pre_max[0] = height[0]
        # for i in range(1, n):
        #     pre_max[i] = max(height[i], pre_max[i - 1])
        
        # suf_max = [0] * n  # suf_max[i] 表示从 height[i] 到 height[n-1] 的最大值
        # suf_max[-1] = height[-1]
        # for i in range(n - 2, -1, -1):
        #     suf_max[i] = max(height[i], suf_max[i + 1])
        
        # for h, pre, suf in zip(height, pre_max, suf_max):
        #     ret += min(pre, suf) - h  # 累加每个水桶能接多少水
        
        # return ret


        # 时间复杂度：O(n)
        # 空间复杂度：O(1)
        # 标签：相向双指针
        n = len(height)
        ret = 0
        left, right = 0, n - 1
        pre_max, suf_max = 0, 0
        while left < right:
            pre_max = max(pre_max, height[left])
            suf_max = max(suf_max, height[right])
            if pre_max < suf_max:
                ret += pre_max - height[left]
                left += 1
            else:
                ret += suf_max - height[right]
                right -= 1
        return ret