给定 n
个非负整数表示每个宽度为 1
的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 1:
输入:height = [0,1,0,2,1,0,1,3,2,1,2,1] 输出:6 解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
示例 2:
输入:height = [4,2,0,3,2,5] 输出:9
按照列方向计算,只要记录左边柱子的最高高度和右边柱子的最高高度,就可以计算出当前位置雨水的面积;当前位置的雨水面积=[min(左边柱子的最高高度,右边柱子的最高高度)-当前柱子高度]x1
使用双指针来遍历,每到一个柱子都向两边遍历一遍,会有重复计算,我们将每个位置左边最高高度记录在一个数组中,右边最高高度记录在一个数组中
当前位置左边最高高度是前一个位置左边最高高度和本高度比较后的最大值
-
从左向右:maxLeft[i]=max(height[i],maxLeft[i])
-
从右向左:maxRight[i]=max(height[i],maxRight[i+1])
class Solution {public int trap(int[] height) {int len=height.length;if(len<=2) return 0;int[] maxLeft=new int[len];int[] maxRight=new int[len];
maxLeft[0]=height[0];for(int i=1;i<len;i++){maxLeft[i]=Math.max(height[i],maxLeft[i-1]);}
maxRight[len-1]=height[len-1];for(int i=len-2;i>=0;i--){maxRight[i]=Math.max(height[i],maxRight[i+1]);}
int sum=0;for(int i=0;i<len;i++){int count=Math.min(maxLeft[i],maxRight[i])-height[i];if(count>0) sum+=count;}return sum;}
}