LeetCode 42：接雨水 动态编程

给定 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

一个柱子当且仅当它左右的柱子的最高高度大于它时，才能接住雨水，那么我们使用两个数，记录左右最大高度即可
lheight[i]表示下标 0 到 i-1 的最大柱子高度
rheight[i]表示下标 i+1 到结尾的最大柱子高度

遍历柱子，只要满足左右高度都大于当前柱子高度，加入答案

代码

class Solution {
public:
    int trap(vector<int>& height)
    {
        if(height.size()<3) return 0;
        vector<int> lheight(height.size());
        vector<int> rheight(height.size());
        for(int i=1; i<height.size(); i++)
            lheight[i] = max(lheight[i-1], height[i-1]);
        for(int i=height.size()-2; i>=0; i--)
            rheight[i] = max(rheight[i+1], height[i+1]);
        int ans = 0;
        for(int i=0; i<height.size(); i++)
            if(lheight[i]>=height[i] && rheight[i]>=height[i])
                ans += min(lheight[i], rheight[i])-height[i];
        return ans;
    }
};