Trapping Rain Water

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


你可能感兴趣的:(Trapping Rain Water)