给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
思路:一块柱子能接水的量取决于它左右两边最高的柱子中较短的一个。从前往后开始遍历各个柱子,只要遇到后面的柱子比前面的水盆边界柱子高,发现就可以构成一个水盆。但是当我们到达最高点之后,后面的就无法算了,因为只有当一个柱子,比它后面所有的柱子都高的时候,才是水盆边界,而又无法确定当前柱子是否比它后面所有的柱子都高。
所以,我们进行两次for循环,第一次找到最高的柱子,这样对于它左边的柱子相当于确定了右边的最高值(水盆边界),对于它右边的柱子相当于确定了左边的最高值(水盆边界),以最高的柱子的位置为分割点,将数组分成两部分。
对于左半部分,不断往右逼近,当一个柱子比前面的水盆边界柱子高,就构成一个水盆。对于右半部分,不断往左逼近,当一个柱子比后面的水盆边界柱子高,就构成一个水盆。累加求所有水盆水量和。
class Solution {
public:
int trap(vector<int>& height) {
if(height.empty()) return 0;
int flag=0, re=0, aa=height[0], bb=height.back();
for(int i=1; i<height.size(); ++i){
if(height[i]>height[flag]) flag=i;
}
for(int i=1; i<=flag; ++i){
if(height[i]<aa)
re+=aa-height[i];
else
aa=height[i];
}
for(int i=height.size()-1; i>=flag; --i){
if(height[i]<bb)
re+=bb-height[i];
else
bb=height[i];
}
return re;
}
};