LeetCode 接雨水（动态规划）

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

解法：

设置sum=0用来保存雨水数。

初始状态：首先用两个数组，max_left [i] 代表第 i 列左边最高的墙的高度，max_right[i] 代表第 i 列右边最高的墙的高度。（一定要注意下，第 i 列左（右）边最高的墙，是不包括自身的）。max_left[0]=0,max_right[0]=0

转移方程：对于 max_left我们其实可以这样求。max_left [i] = Max(max_left [i-1],height[i-1])。它前边的墙的左边的最高高度和它前边的墙的高度选一个较大的，就是当前列左边最高的墙了。对于 max_right我们可以这样求。max_right[i] = Max(max_right[i+1],height[i+1]) 。它后边的墙的右边的最高高度和它后边的墙的高度选一个较大的，就是当前列右边最高的墙了。

计算顺序：从左到右，依次比较height[i]与max_left[i]和max_right[i]中的最小值temp，若最小值大于height[i]，则max=max+temp-height[i]。

代码如下：

class Solution {
public:
    int trap(vector<int>& height) {
        int sum=0;
        int len=height.size();
        vector<int>left(len,0);
        vector<int>right(len,0);
        for(int i=1;i<len;i++){
            left[i]=max(height[i-1],left[i-1]);
        }
        for(int i=len-2;i>=0;i--){
            right[i]=max(height[i+1],right[i+1]);
        }
        for(int i=1;i<len-1;i++){
            int temp=min(left[i],right[i]);
            if(temp>height[i])
                sum+=(temp-height[i]);
        }
    return sum;
    }
};

结果如下：