leetcode-42接雨水问题

题目

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

示例

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

代码

class Solution {
    public static int trap(int[] height) {
    if(height.length<=2){return 0;}
    int result=0;
    int left_max=height[0];
    int right_max=height[height.length-1];
    //这道题目的关键是建立两个数组
    //height1[i]保存i位置左边最大的高度
    //height2[i]保存i位置右边最大的高度
    int[]height1=new int[height.length];
    int[]height2=new int[height.length];
    for(int i=1;i<height.length;i++){
        if(height[i]>=left_max){
            left_max=height[i];
        }
         height1[i]=left_max;
    }
     for(int i=height.length-2;i>=0;i--){
        if(height[i]>=right_max){
            right_max=height[i];
        }
         height2[i]=right_max;
    }
    for(int i=1;i<height.length-1;i++){
    //从第二个位置一直遍历到倒数第二个位置
    //因为最后一个和第一个位置是不能存储水的
    //取当前位置的高度和当前位置左边最大值和右边最大值两者的最小值进行比较
    //如果大于这两者最小值表示无法存储水
        int length=Math.min(height1[i],height2[i]);
        if(length>height[i]){
            result=result+length-height[i];
        }
    }
    return result;
    }
}