对于数组中的每个元素,我们找出下雨后水能达到的最高位置,等于两边最大高度的较小值 减去当前高度的值 就能的得到当前的存水量。
在暴力方法中,我们仅仅为了找到最大值每次都要向左和向右扫描一次。但是我们可以提前存储这个值。因此,可以通过动态编程解决。
找到数组中从下标 i 到最左端最高的条形块高度 left[i]。 找到数组中从下标 i 到最右端最高的条形块高度 right[i]。 扫描数组 height 并更新答案: 即累加 min(left[i],right[i])−height[i]到 num上
public static int trap(int[] height) {
int len=height.length;//暂存数组长度
int num=0; //存储储水量
if(height==null || len==0){
return 0;
}
int []left=new int [len];//存储往左找的最大值
int []right=new int [len];//存储往右找的最大值
left[0]=height[0]; //第一个往左找的最大值就是他本身
right[len-1]=height[len-1]; //最后一个往右找的最大值就是他本身
for(int i=1;i<len;i++){ //从第二个开始找
left[i]=Math.max(height[i],left[i-1]);
}
for(int i=len-2;i>=0;i--){ //从倒数第二个开始找
right[i]=Math.max(height[i],right[i+1]);
}
for(int i=1;i<len-1;i++){ //循环height数组,给num累加值
num=num+ Math.min(left[i],right[i])-height[i];
}
return num; //返回结果
}