题目
给定n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
示例:
图略,
其中所给数组是柱子的高度
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
分析
1.首先使用双指针法,这样可以一次搞定。如果左边元素值比右边元素值小,则从左边元素开始计算。
反之,则从右边元素开始计算。同时需要左右各维护一个最大值,如果当前值比最大值小,则最大值减去
当前值,然后加入res中。否则就更新最大值。
2.之所以比他小更新,是因为只有当前值比前面得出的最大值小才可以接雨水,大的话就不行,只能作为
更新最大值,这个需要仔细观察题目所给信息
附注:
1.有序数组可以考虑用双指针法。
2.观察题目
#include<iostream>
#include<vector>
using namespace std;
class MyClass
{
public:
int trap(vector<int>& height) {
if (height.empty()){
return 0;
}
int right = height.size()-1, left = 0;
int maxRight = 0, maxLeft = 0;
int res = 0;
while (left < right){
if (height[left] < height[right]){
height[left] >= maxLeft ? maxLeft = height[left] : res += maxLeft - height[left];
++left;
}
else
{
height[right] >= maxRight ? maxRight = height[right] : res += maxRight - height[right];
--right;
}
}
return res;
}
};