LeetCode 42. Trapping Rain Water

题意

给一个序列，里面是柱子的高度，求在一场雨之后，可以存储多少水
例：

[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;
    }
};