算法 - leetcode 42. 接雨水

class Solution {
    public int trap(int[] height) {
        if(height==null || height.length<=2){
            return 0;
        }

        int total=0;
        int left=0;

        //左低右高
        for(int i=0;i<height.length;i++){
            if(height[left]<=height[i]){//计算赋值
                //判断是否符合"凹"型（可以装水）
                if(left+2<=i && height[left]>height[left+1] && height[i]>height[i-1]){
                    int tmpTotal=0;
                    for(int j=left+1;j<i;j++){
                        tmpTotal+=height[j];
                    }
                    //积水量=补平最高位后的总量  - 台阶占用空间
                    total=total+(height[left]*(i-left-1) - tmpTotal);
                }

                left=i;
            }
        }


        //左高右低
        int right=height.length-1;
        for(int i=right;i>=left;i--){
            if(height[right]<=height[i]){//计算赋值
                if(i+2<=right && height[i]>height[i+1] && height[right]>height[right-1]){
                    int tmpTotal=0;
                    for(int j=i+1;j<right;j++){
                        tmpTotal+=height[j];
                    }

                    total=total+(height[right]*(right-i-1) - tmpTotal);
                }

                right=i;
            }


        }



        return total;
    }
}