42. 接雨水

题目

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

思路

• 可以观察，对于任意一个高度，如果它的两边区间分别存在两个最高的高度，且都高于当前的高度，则当前高度可以装水，量为较小的减去当前高度。
• 因为对于每一个高度，需要知道两边最高的高度是多少，所以使用两个数组，正反遍历一遍。一个装每个元素左边的最高高度，一个装右边的最高高度。
• 然后再遍历一遍按照1的策略来装水。

代码

class Solution {
public:
    int trap(vector<int>& height) {
        int lens = height.size();

        if ( lens < 3 ) return 0;

        vector<int> maxLeft( lens, 0 );     // 每个数字左边区间里最大的数字
        vector<int> maxRight( lens, 0 );    // 每个数字右边区间里最大的数字

        maxLeft[1] = height[0];
        for ( int i = 2; i < lens; ++i ) {
            maxLeft[i] = max( maxLeft[i-1], height[i-1] );
        }

        maxRight[lens - 2] = height[lens-1];
        for ( int i = lens - 3; i >= 0; --i ) {
            maxRight[i] = max( maxRight[i+1], height[i+1] );
        }

        int res = 0;

        for ( int i = 1; i < lens - 1; ++i ) {
            int minVal = min( maxLeft[i], maxRight[i] );
            
            if ( minVal > height[i] )       // 如果当前数字两边的最高的数都大于它
                res += minVal - height[i];  
        }

        return res;
    }
};