接雨水

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

 

 

class Solution {
public:
    int trap(vector<int>& height) 
    {
#if 0
        //解法1   
        if(height.empty())
            return 0;
        int size = height.size();
        int ret = 0;
        vector<int> l_max(size);
        vector<int> r_max(size);
        l_max[0] = height[0];
        r_max[size-1] = height[size-1];
        for(int i = 1; i < size; i++)
        {
            l_max[i] = max(l_max[i-1], height[i]);
        }
        
        for(int i = size - 2; i >= 0; i--)
        {
            r_max[i] = max(r_max[i+1], height[i]);
        }
        
        for(int i = 1; i < size - 1; i++)
            ret += min(r_max[i], l_max[i]) - height[i];
        
        return ret;
#endif
        
        //解法二
        if(height.empty())
            return 0;
        int size = height.size();
        int left = 0;
        int right = size - 1;
        int ret = 0;
        int l_max = height[left];
        int r_max = height[right];
        while(left <= right)
        {
            l_max = max(l_max, height[left]);
            r_max = max(r_max, height[right]);
            if(l_max < r_max)
            {
                ret += l_max - height[left];
                left++;
            }
            else
            {
                ret += r_max - height[right];
                right--;
            }
        }
        
        return ret;
    }
};