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

class Solution {
public:
    int trap(vector& height) {
        if(height.size()<3)
            return 0;
        int localMax = height[0];
        vector left(height.size());
        vector right(height.size());
        for(int i=0;i             if(height[i] <= localMax)
                left[i] = localMax;
            else {
                localMax = height[i];
                left[i] = localMax;
            }
        }
        
        localMax = height[height.size()-1];
        for(int i=height.size()-1;i>-1;i--) {
            if(height[i] <= localMax)
                right[i] = localMax;
            else {
                localMax = height[i];
                right[i] = localMax;
            }
        }
        int area = 0;
        for(int i=0;i         {
            area += min(left[i], right[i]) -height[i];
        }
        return area;
    }
};

你可能感兴趣的:(leetcode中国)