题干
给定 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));
}
}