题目:Trapping Rain Water
原题介绍
原题链接:https://leetcode.com/problems/trapping-rain-water/#/description
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.
给出一组整数序列,代表台阶的高度,每个台阶的宽度都是1,计算一共能蓄水多少。
例:[0,1,0,2,1,0,1,3,2,1,2,1]
如上图,蓝色部分就是模拟的蓄水的部分,总计可以蓄水的值是6.
思路
用两个指针i和j,初始i指向序列的最左边,j指向序列的最右边,然后开始扫描,每次从矮的一边开始往中间扫描,用maxL和maxR分别记录左边的最大值和右边的最大值,当左边开始扫描时,如果遇到的值比左边的最大值小,那么这一块就可以蓄水maxL - height[ i ]的量,当遇到的值比maxL大时,就更新maxL,右边同理,一直遍历到两个指针相遇。
为什么要从矮的一边开始呢,类似于木桶原理,左右两边高度不一的话,肯定水量最高只能是矮边的高度。
参考代码
class Solution {
public:
int trap(vector<int>& height) {
int len = height.size();
int i = 0, j = len - 1, maxL = 0, maxR = 0, ans = 0;
while(i <= j) {
if(height[i] <= height[j]) {
if(height[i] > maxL) maxL = height[i];
else {
ans += maxL - height[i];
}
i++;
}else{
if(height[j] > maxR) maxR = height[j];
else {
ans += maxR - height[j];
}
j--;
}
}
return ans;
}
};