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