求每一列的水,我们只需要关注当前列,以及左边最高的墙,右边最高的墙就够了。装水的多少,我们只需要看左边最高的墙和右边最高的墙中较矮的一个就够了。
遍历每一列,然后分别求出这一列两边最高的墙。找出较矮的一端,和当前列的高度比较,如果当前列比较矮的一端高,那么不会注水。如果当前的列比较矮的一列矮,那么注水的深度就是较矮的那一列减去当前这一列。
我们可以使用动态规划,创建两个数组分别保存左边最高的列和右边最高的列,这样就不用每次都进行一次遍历求最高的列。
var trap = function(height) {
var max_left=[];//定义一个数组,存放左边的最小值
var max_right=[];//定义一个数组,存放右边的最小值
var min,sum=0;
//如果不对它们进行赋值则会出现无法比较的情况
max_left[0]=0;
max_right[height.length-1]=0;
//从左边开始依次寻找最小值
for(let i=1;i<height.length-1;i++){
max_left[i]=Math.max(max_left[i-1],height[i-1]);
}
//从右边开始依次寻找最大值
for(let i=height.length-2;i>=0;i--){
max_right[i]=Math.max(max_right[i+1],height[i+1]);}
//如果当前的柱子高度低于左边和右边最大值的最小值,则可以接水
for(let i=1;i<height.length-1;i++){
min=Math.min(max_left[i],max_right[i]);
if(height[i]<min){
sum=sum+min-height[i];
}
}
return sum;
};