【动态规划】接雨水

leetcode链接https://leetcode-cn.com/problems/trapping-rain-water/
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

如下图，本题重要的不是动态规划的思想，而是几何思维解题的方法

• step1.从左向右遍历找到最大
• step2.从右向左遍历找到最大
• step3.加起来，并删去多余的部分

定义：dp_left[i]表示从左向右当前最大值，dp_right[i]表示从右向左当前最大值

边界：可不定义边界，初始化为dp_left[0]=0，dp_right[N-1]=0

状态转移方程：dp_left[i] = max(dp_left[i-1],nums[i]) dp_right[i] = max(dp_right[i+1],nums[i])

class Solution:
    def trap(self, height: List[int]) -> int:
        # 边界条件
        if not height: return 0
        n = len(height)
        maxleft = [0] * n
        maxright = [0] * n
        ans = 0
        # 初始化
        maxleft[0] = height[0]
        maxright[n-1] = height[n-1]
        # 设置备忘录，分别存储左边和右边最高的柱子高度
        for i in range(1,n):
            maxleft[i] = max(height[i],maxleft[i-1])
        for j in range(n-2,-1,-1):
            maxright[j] = max(height[j],maxright[j+1])
        # 一趟遍历，比较每个位置可以存储多少水
        for i in range(n):
            if min(maxleft[i],maxright[i]) > height[i]:
                ans += min(maxleft[i],maxright[i]) - height[i]
        return ans

作者：z1m
链接：https://leetcode-cn.com/problems/trapping-rain-water/solution/dong-tai-gui-hua-shuang-zhi-zhen-tu-jie-by-ml-zimi/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

代码优化（一次遍历，优化空间，思路跟最大子序列问题一致）：

class Solution:
    def trap(self, height: List[int]) -> int:
        Left_h,Right_h = 0 ,0
        N   = len(height)
        ans = 0
        for i in range(N):
            Left_h = max(Left_h, height[i])
            Right_h= max(Right_h,height[N-i-1])
            ans   += Left_h + Right_h - height[i]
        return ans - N*Left_h