Trapping Rain Water

最新推荐文章于 2024-09-17 16:23:14 发布

原创最新推荐文章于 2024-09-17 16:23:14 发布 · 410 阅读

0 ·

CC 4.0 BY-SA版权

算法数据结构专栏收录该内容

93 篇文章

订阅专栏

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

思路：首先找到最高的那个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;
    }