Trapping Rain Water

建立一个数组num,表示当前位置的水平高度(包括水和bar之和)。第一个位置num[0]和最后一个位置num[n-1]因为是开区间,肯定不能trap water。

从第二个位置num[1]开始往后扫描,如果num[i]取num[i-1]和A[i]的较大者。表示该位置水平高度的上限(尚未考虑右侧邻居)。

然后从倒数第二个位置开始往前扫描,保证num[i] <= num[i+1],并且num[i]不小于A[i]。

每个位置trap的水量即为num[i]-A[i]。

时间O(n),空间O(n)。

class Solution {
public:
    int trap(int A[], int n) {
        int *num = new int[n];
        memset(num, 0, sizeof(int)*n);
        int total = 0;
        
        num[0] = A[0];
        num[n-1] = A[n-1];
        
        for(int i=1; i<n-1; i++)
            num[i] = max(A[i], num[i-1]);
        
        for(int i=n-2; i>=1; i--)
        {
            num[i] = max(A[i], min(num[i], num[i+1]));
            total += (num[i]-A[i]);
        }
        
        delete []num;
        return total;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值