Leetcode 42. 接雨水

Leetcode 42. 接雨水

题目说明

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

题目解析

1. 动态规划
   遍历数组两次，得到每个坐标的左边最大高度和右边最大高度数组
   遍历数组，若坐标的左右最大高度都不为0且都比该坐标的高度大，则返回的结果 + 左右最大高度的最小值 - 当前坐标的高度
   返回结果

Python代码

class Solution:
    def trap(self, height: List[int]) -> int:
        n_max, length, result = 0, len(height), 0
        l_max, r_max = [0], [0]
        for i in range(1, length):
            n_max = n_max if n_max >= height[i - 1] else height[i - 1]
            l_max.append(n_max)
        n_max = 0
        for i in range(length - 2, -1, -1):
            n_max = n_max if n_max >= height[i + 1] else height[i + 1]
            r_max.append(n_max)
        for i in range(length):
            if l_max[i] and r_max[length - 1 - i]:
                m = min(l_max[i], r_max[length - 1 - i])
                if m > height[i]:
                    result += m - height[i]
        return result