leetcode【高级】接雨水 Java

题干

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

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

示例:

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

想法

首先我们使用数组int[]left来维护这个位置左边最高的高度
int[]right同理
所以:
[0,1,0,2,1,0,1,3,2,1,2,1] <–柱子高度
[0,0,1,1,2,2,2,2,3,3,3,3] <–左木板，等于前一个柱子的高度与之前左木板的最大值的最大值
[3,3,3,3,3,3,3,2,2,2,1,0] <–右木板 ，等于后一个柱子的高度与之后的有木板的最大值的最大值

维护完这个以后再遍历一次数组，如果这个位置的left和right都大于height
那么这一个位置的蓄水量就是left和right里小的那个再➖height

直接看代码吧很好懂

java代码

class Solution {
    public int trap(int[] height) {
        int result=0;//最终结果
        int len=height.length;
        int[]left=new int[len];//左数组维护前一个柱子的高度与之前左木板的最大值的最大值
        int[]right=new int[len];
        if(len<=2){
            return 0;
        }
        
        for(int i=1;i<len;i++){
            left[i]=Math.max(left[i-1],height[i-1]);//遍历一遍得到left，下痛
            right[len-i-1]=Math.max(right[len-i],height[len-i]);
        }
        for(int j=0;j<len;j++){
        if(left[j]>height[j]&&right[j]>height[j]){//result等于小的那个减height
            result+=Math.min(left[j],right[j])-height[j];      
        }
        }
        return result;
    }
}