题目链接: https://leetcode.com/problems/trapping-rain-water/
Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.
For example,
Given [0,1,0,2,1,0,1,3,2,1,2,1]
, return 6
.
The above elevation map is represented by array [0,1,0,2,1,0,1,3,2,1,2,1]. In this case, 6 units of rain water (blue section) are being trapped. Thanks Marcos for contributing this image!
思路: two point指向左右两端, 维护一个当前左右同时能达到的最大的高度, 并且让面积小的一端先移动, 如果遇到一个新的高度小于当前最高高度, 那么其高度差就可以存储一些水. 如果碰到一个高度, 使得左右能同时达到的高度增加, 则更新.
代码如下:
class Solution {
public:
int trap(vector<int>& height) {
int left = 0, right = height.size()-1, ans =0, Min =0;
while(left <= right)
{
int val = min(height[left], height[right]);
if(val < Min) ans += (Min - val);
else Min = val;
height[left]<=height[right]?left++:right--;
}
return ans;
}
};