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

想法

这道题之前做过
想法是这样的
使用left 用来存储从现在的位置到左边最高的高度
right同理
因为接的雨水的多少其实取决于两边的最小值减去中间原本height的高度（如果height比两边小的话
然后再遍历一遍使用小的那一个减去height的高度即可

Java代码

package daily;

public class Trap {
    public int trap(int[] height) {
         /*
          * @Description:leftright分别维护向左的最高值和向右的最高值
          * @Author: ragezor
          * @Date: 2020/4/4 6:17 下午
          * @Param [height]
          * @return int
          **/
        int len=height.length;
        int [] left=new int[len];
        int[] right=new int[len];
        int res=0;
        //维护leftright
        for (int i=1;i<len;i++){
            left[i]=Math.max(left[i-1],height[i-1]);
            right[len-1-i]=Math.max(right[len-i],height[len-i]);
        }
        //再遍历一遍用两边的小值减去height（如果height更低的话
        for(int j=0;j<len;j++){
            if(left[j]>height[j]&&right[j]>height[j]){
                res+=Math.min(left[j],right[j])-height[j];
            }
        }

        return res;
    }
    public static  void main(String[] args){
        Trap trap=new Trap();
        int [] test={0,1,0,2,1,0,1,3,2,1,2,1};
        System.out.println(trap.trap(test));
    }
}

我的leetcode代码都已经上传到我的git