Trapping Rain Water

本文介绍了一种计算容器内水体积的算法,通过三次循环实现,复杂度为O(n)。算法首先找到最高bar,然后从两端向中间最高bar遍历,计算每个bar能存储的水量。注意数组长度小于3时返回0。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

思路:首先找到最高的那个bar,然后分别从左右两端往中间最高的bar遍历。3个for循环,复杂度为O(n)。

左边往中间遍历时,遍历到第i个位置时,这个bar能存储的水量是leftmax-A[i],leftmax是之前遍历的bar(包括第i个位置)中最高的那个值。

同理,右边往中间遍历时也是如此。

注意当数组长度小于3时,return 0.

public int trap(int[] A) {
       int n=A.length;
       if(n<3)return 0;
       int maxIndex = 0,maxValue=0;
       int sum =0;
       for(int i=0;i<n;i++){
           if(maxValue<A[i]){
              maxIndex = i;
              maxValue = A[i];
           }
       }
       int left=1,right=n-2;
       int leftmax=A[0],rightmax=A[n-1];
       for(int i=1;i<maxIndex;i++){
           if(A[i]>leftmax){
               leftmax = A[i];
           }
           sum += leftmax-A[i];
       }
       for(int i=n-2;i>maxIndex;i--){
           if(A[i]>rightmax){
               rightmax = A[i];
           }
           sum += rightmax-A[i];
       }
       return sum;
    }


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值