题目描述:给定n个非负整数表示高地,每个高地的宽度是1,计算下雨后由这些高地能储存多少水?
例子:
事实一:储水其实是一个个孤立的单位,特点是最高的两个柱子刚好是两个边上的柱子
事实二:一个水柱的体积
V
=
(
h
−
h
′
)
∗
w
V=(h-h')*w
V=(h−h′)∗w
h等于水注的高度,h’是该位置柱子的高度,w=1
水柱的高度,其实是两个边界上较短的那根柱子的高度(所谓的短板)
所以边界中短板减去中间储水的柱子的高度,得到的就是每个水柱的体积
事实三:如果height[i]处往右可以积水,那么位置i+1 一定低于位置i
从右边看,如果位置j往左能积水,那么一定有j-1低于位置j。
基于上面三件事,我们计算体积的过程就是找到一个积水单位(事实三提供特征),计算这个积水单位的积水体积(事实二提供特征)。
接下来,怎么知道某个积水单位的较短的边界呢?
很巧妙的是,假如我们找到了height[i]处往右边可以积水,height[j]处往左边可以积水(显然j>i)那么,height[i]和height[j]中比较小的一个,一定是一个短板(水柱边界且高度较小的一个)举一个例子:
4 , 3 , 5 , 2 , 7 4,3,5,2,7 4,3,5,2,7
这个搭配里,我们发现4往右可以积水,7往左可以积水,4<7,那么4一定是某个积水单位的较短边。因为假如存在一个比4更短的边那么水会继续往右流,直到7挡住水流。从右往左分析类似。
综上,这个算法的流程就是:
(1)找到左右两边可以积水的地方,也就是满足
h
e
i
g
h
t
[
i
]
>
h
e
i
g
h
t
[
i
+
1
]
height[i]>height[i+1]
height[i]>height[i+1]
h
e
i
g
h
t
[
j
]
>
h
e
i
g
h
t
[
j
−
1
]
height[j]>height[j-1]
height[j]>height[j−1]
的i和j的位置
(2)然后找到height[i]和height[j]中较短的一个,这个一定是某个积水单位的较短板,然后逐个单位去减去中间较短的长度得到的就是这个水柱的体积
(3)求完一个积水单位后,继续找下一个积水单位,直到两个边界也就是i和j走到一起,算法结束
代码:
class Solution {
public int trap(int[] height) {
int len = height.length;
if(height==null||len<=2)return 0;
int left = 0;
int result = 0;
int right = len-1;
while(left<right && height[left]<height[left+1])left++;
while(left<right && height[right]<height[right-1])right--;
while(left<right){
int lh = height[left];
int rh = height[right];
if(lh<rh){
while(left<right && height[++left]<lh){
result+=lh-height[left];
}
}else{
while(left<right&&height[--right]<rh){
result+=rh-height[right];
}
}
}
return result;
}
}
时间复杂度:O(n)