1、题目
给定 n 个非负整数表示每个宽度为 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、题目分析
“接雨水”算是一类比较经典的题目,解这类题最重要的是看清题目的本质。我刚刚接触到这个题目的时候,会把整个系统分为泥土和水两个部分,但是一想泥土的高度是随机的,水的体积也很混乱,很难求出来。后来我发现把泥土和水看做一个整体的话,题目就简单多了,这时候我们就会发现整个图形是一个梯形,两边低中间高!
认识到这是个梯形之后,下一个就很很简单了。水位是一层一层的,只需要从两边逐渐向中间靠拢,水位是逐渐上升的,每一个位置的水位可以由大梯形这一层泥土的高度得到。
3、具体步骤
设置两个位置变量,一个是 left,一个是 right,分别代表左右推进到哪儿了。为了省事和准确,可以一次只移动一个方向的一格(左或者右),直到 left 和 right 相遇,总雨水量就得到了。
注意:每次移动时,选择更低的那边移动。最终才会向最高点靠拢。
4、代码(java)
public int trap(int[] height) {
int left=0,right=height.length-1,ans=0;
int hl=0, hr=0;//代表左右两边分别到达的层的层高
while (left <= right){
//选择更低的那边移动
if(height[left]<=height[right]){
if(height[left] <= hl) ans += (hl-height[left]);
else hl = height[left];
left++;
}
else{
if(height[right] <= hr) ans += (hr-height[right]);
else hr = height[right];
right--;
}
}
return ans;
}
5、总结
该算法并不难理解,就是把整体看成一个梯形,一层一层的求即可。