NC128 接雨水问题

NC128 接雨水问题

描述
给定一个整形数组arr，已知其中所有的值都是非负的，将这个数组看作一个柱子高度图，计算按此排列的柱子，下雨之后能接多少雨水。
示例1
输入：
[3,1,2,5,2,4]
复制
返回值：
5
复制
说明：
数组 [3,1,2,5,2,4] 表示柱子高度图，在这种情况下，可以接 5个单位的雨水，蓝色的为雨水
示例2
输入：
[4,5,1,3,2]
复制
返回值：
2

import java.util.*;


public class Solution {
    /**
     * max water
     * @param arr int整型一维数组 the array
     * @return long长整型
     */
    public long maxWater (int[] arr) {
        // write code here
        if(arr == null || arr.length == 0){
            return 0 ;
        }
        int len = arr.length ;
        int[] left = new int[arr.length] ;
        int[] right = new int[arr.length] ;
        left[0] = arr[0] ;
        right[len-1] = arr[len-1] ;
        for(int i = 1 , j = len -2 ; i < len && j>= 0 ; i++ , j--){
            left[i] = Math.max(left[i-1] , arr[i]) ;
            right[j] = Math.max(right[j+1] , arr[j]) ;
        } 
        long ans = 0 ;
        for(int i = 0 ; i < len ; i++){
            ans = ans + Math.min(right[i] , left[i]) - arr[i] ;
        }
        return ans ;
        
    }
}