20200404

接雨水

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

上面是由数组 [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

int trap(vector<int>& height) {
    int n = height.size();
    // left[i]表示i左边的最大值，right[i]表示i右边的最大值
    vector<int> left(n), right(n);
    for(int i = 1; i < n; i++){
        left[i] = max(left[i - 1], height[i - 1]);
    }
    for(int i = n - 2; i >= 0; i--){
        right[i] = max(right[i + 1], height[i + 1]);
    }
    int water = 0;
    for(int i = 0; i < n; i++){
        int level = min(left[i], right[i]);
        water += max(0, level - height[i]);
    }
    return water;
}