42. Trapping Rain Water

这里写图片描述
我的第一想法是找到数组的最大值,然后分左右两边求能装的水量,再加起来。

class Solution {
public:
    int trap(vector<int>& height) 
    {
        if(!height.size()||height.size()==1)
            return 0;
        int i,j,max=0,index,width;
        for(i=0;i<height.size();i++)
        {
            if(height[i]>max)
            {
                max=height[i];
                index=i;
            }
        }
        i=0;
        int start,end;
        start=i;
        int temp,sum;
        temp=0;sum=0;
        for(j=i+1;j<index;j++)
        {
            if(height[j]>height[start])
            {
                end=j;
                width=end-start-1<0?0:end-start-1;
                sum=sum+width*height[start]-temp;
                temp=0;
                start=end;
            }
            else
                temp=temp+height[j];    
        }
        end=index;
        width=end-start-1<0?0:end-start-1;
        sum=sum+width*height[start]-temp;
        i=height.size()-1;
        start=i;
        temp=0;
        for(j=i-1;j>index;j--)
        {
            if(height[j]>height[start])
            {
                end=j;
                width=start-end-1<0?0:start-end-1;
                sum=sum+width*height[start]-temp;
                temp=0;
                start=end;
            }
            else
                temp=temp+height[j];
        }
        end=index;
        width=start-end-1<0?0:start-end-1;
        sum=sum+width*height[start]-temp;
        return sum;
    }
};

然后看别人的想法采用l和r两个指针,维护装水两边的位置。
当l处高度低时,右移l,同是加上l处与右移后位置高度差(因为这里都能装水啊),直到再遇到同样高或者更高的位置。然后进行下一轮判断。
同样,当r处高度低时,左移r,同是加上r处与左移后位置高度差,直到再遇到同样高或者更高的位置。
最后直到l和r相遇,结束。

class Solution {
public:
    int trap(vector<int>& height) 
    {
        int l=0,
            r=height.size()-1,
            lheight=height[l],
            rheight=height[r],
            water=0;
        while(l<r)
        {
            if(lheight<rheight)
            {
                l++;
                if(height[l]<lheight)
                    water+=lheight-height[l];
                else
                    lheight=height[l];
            }
            else
            {
                r--;
                if(height[r]<rheight)
                    water+=rheight-height[r];
                else
                    rheight=height[r];
            }
        }
        return water;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值