LeetCode42. 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.
思路
刚开始思路是一层一层的统计,找到两个相邻的柱子,统计他们的间距,中间加上一些条件,后来虽然实现了但果然超时了,不过实现了我已经很满意了。
网上看到的比较好的思路是两边往中间遍历,记录当前第二高点H,然后利用这个第二高点减去当前历经的柱子,剩下就装水容量了。(为什么是第二高点?因为两边比较,最高的点不用动,只移动第二高点,这样可以保证水不会”溢出“。)
代码实现
class Solution {
public:
int trap(vector<int>& height) {
int n=height.size();
if(n==0) return 0;
int left=0;
int right=n-1;
int H=0;
int water=0;
while(left<right){
if(height[left]<height[right]){
H=max(height[left],H);
water+=H-height[left];
left++;
}else{
H=max(height[right],H);
water+=H-height[right];
right--;
}
}
return water;
}
};
ps: 不理解LeetCode为什么把这题放在stack标签里,难道是用stack有更好的解法?