第一种方法:双指针
利用左右指针的下标差值计算出每一层雨水和柱子的体积。如下图,第一层体积为11,第二层为8,第三层为1。累加得到整体体积area= 20(注意:每一层从左边第一个方格到右边最后一个方格之间一定是被蓝黑两种颜色的方格填满,不会存在空白,所以我们可以这么求值)
计算柱子的总体积Sum,就是height:[0,1,0,2,1,0,1,3,2,1,2,1]数组之和14
返回结果area− Sum就是雨水的体积。
class Solution {
public:
int trap(vector<int>& height) {
int sum = accumulate(height.begin(), height.end(), 0);
int area = 0;
int floor = 1, left = 0, right = height.size() - 1;
while (left <= right) {
while (left <= right && height[left] < floor) {
left++;
}
while (left <= right && height[right] < floor) {
right--;
}
area += (right - left + 1);
floor += 1;
}
return area - sum;
}
};
方法二:单调栈
除了计算并存储每个位置两边的最大高度以外,也可以用单调栈计算能接的水的总量。
维护一个单调栈,单调栈存储的是下标,满足从栈底到栈顶的下标对应的数组 height 中的元素递减。
从左到右遍历数组,遍历到下标 ii 时,如果栈内至少有两个元素,记栈顶元素为 top,top 的下面一个元素是 left,则一定有 height[left]≥height[top]。如果 height[i]>height[top],则得到一个可以接雨水的区域,该区域的宽度是i−left−1,高度是min(height[left],height[i])−height[top],根据宽度和高度即可计算得到该区域能接的水的量。
为了得到left,需要将top 出栈。在对 top 计算能接的水的量之后,left 变成新的 top,重复上述操作,直到栈变为空,或者栈顶下标对应的height 中的元素大于或等于height[i]。
在对下标 i 处计算能接的水的量之后,将 i 入栈,继续遍历后面的下标,计算能接的水的量。遍历结束之后即可得到能接的水的总量。
class Solution {
public:
int trap(vector<int>& height) {
int ans = 0;
stack<int> stk;
int n = height.size();
for (int i = 0; i < n; ++i) {
while (!stk.empty() && height[i] > height[stk.top()]) {
int top = stk.top();//上一个left位置,栈下面的值也一定比top大,且位置在top前,right>left,中间就能形成接水区域
stk.pop();
if (stk.empty()) {
break;
}
int left = stk.top();//left后移,进行计算
int currWidth = i - left - 1;
int currHeight = min(height[left], height[i]) - height[top];//一层一层计算面积
ans += currWidth * currHeight;
}
stk.push(i);
}
return ans;
}
};