题意
给一个序列,里面是柱子的高度,求在一场雨之后,可以存储多少水
例:
[0,1,0,2,1,0,1,3,2,1,2,1]
如图所示,箭头所指即为所能储存的水量
思路
如果一个区间内如果可以存水,那么这个区间必定满足凹函数的分布,所以说可以设置一个指针指向遍历过程中最高的那根柱子,然后如果遇到和最高的柱子一样高或者比其还要高的柱子,那么说明这里可以存水,当然,中间需要记录,这时需要切换指针到这个柱子继续寻找。但是还有一种情况,在后面找不到比指针指向的柱子还要高的柱子,比如5,3,2,4
这种情况,虽然找不到比指针指向的柱子还要高的柱子,但是里面还是可以存水的,对于这种情况,可以再倒着遍历,此时寻找条件应该为寻找比指针指向的柱子要高的柱子,因为两个一样高的情况前面已经遍历过一次了.
代码
class Solution {
public:
int trap(vector<int>& height) {
size_t len = height.size();
int top = -1;//指向最高的那个柱子
int ans = 0;
int val = 0;//中间记录存水量
for(int i = 0; i < len; i++){
if(height[i] < top){
val += top - height[i];
} else{
ans += val;
top = height[i];
val = 0;
}
}
//倒着遍历
top = -1;
val = 0;
for(int i = len - 1; i >= 0; i--){
if(height[i] < top){
val += top - height[i];
} else if(height[i] > top){
ans += val;
top = height[i];
val = 0;
}
}
return ans;
}
};