首先要知道两个柱子间的水量是由短柱子决定的。采用 左右双指针 的方法,同时分别维护一个左右的当前最大柱子值:
1)、如果left_max < right_max,那么从左边开始遍历,并更新left_max以及结果大小;
2)、如果left_max >= right_max,那么从右边开始遍历,并更新right_max以及结果大小。
算法如下:
代码如下:
class Solution {
public:
int trap(vector& height) {
int res = 0;
int l_max = 0, r_max = 0; //维护左右最大值
int l = 0, r = height.size() - 1; //左右“指针”
while(l < r)
{
if(height[l] < height[r]){
height[l] >= l_max ? l_max = height[l] : res += (l_max - height[l]);
l++;
}
else{
height[r] >= r_max ? r_max = height[r] : res += (r_max - height[r]);
r--;
}
}
return res;
}
};