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 

思路:一块柱子能接水的量取决于它左右两边最高的柱子中较短的一个。从前往后开始遍历各个柱子,只要遇到后面的柱子比前面的水盆边界柱子高,发现就可以构成一个水盆。但是当我们到达最高点之后,后面的就无法算了,因为只有当一个柱子,比它后面所有的柱子都高的时候,才是水盆边界,而又无法确定当前柱子是否比它后面所有的柱子都高。

所以,我们进行两次for循环,第一次找到最高的柱子,这样对于它左边的柱子相当于确定了右边的最高值(水盆边界),对于它右边的柱子相当于确定了左边的最高值(水盆边界),以最高的柱子的位置为分割点,将数组分成两部分。

对于左半部分,不断往右逼近,当一个柱子比前面的水盆边界柱子高,就构成一个水盆。对于右半部分,不断往左逼近,当一个柱子比后面的水盆边界柱子高,就构成一个水盆。累加求所有水盆水量和。

class Solution {
public:
    int trap(vector<int>& height) {
        if(height.empty()) return 0;
        int flag=0, re=0, aa=height[0], bb=height.back();
        for(int i=1; i<height.size(); ++i){
            if(height[i]>height[flag]) flag=i;
        }
        for(int i=1; i<=flag; ++i){
            if(height[i]<aa)
                re+=aa-height[i];
            else 
                aa=height[i];
        }
        for(int i=height.size()-1; i>=flag; --i){
            if(height[i]<bb)
                re+=bb-height[i];
            else
                bb=height[i];
        }
        return re;
    }
};

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值