给定 n
个非负整数表示每个宽度为 1
的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
输入:height = [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 个单位的雨水(蓝色部分表示雨水)。
示例 2:
输入:height = [4,2,0,3,2,5] 输出:9
提示:
n == height.length
1 <= n <= 2 * 104
0 <= height[i] <= 105
//遍历配合剪枝
class Solution {
public:
int trap(vector<int>& height) {
if(height.size()==1) return 0;
if(height.size()==2) return abs(height[0]-height[1]);
//先求下标1的柱子上可容纳多少单位雨水
//同时是max_left max_right的初始化
int max_left = height[0];
int max_right = 0;
int vol = 0;
int limit;
for (int i = 2; i < height.size(); i++) {
if (height[i] > max_right)
max_right = height[i];
}
limit = min(max_left, max_right);
if (limit > height[1])
vol = vol + (limit - height[1]);
//每次后移
//左端出现新的元素,与当前边界比较取较大者
//右端边界若与当前位置高度相等,则可能右端边界正是当前位置,需重新寻找
for (int i = 2; i < height.size() - 1; i++) {
if (max_left < height[i - 1])
max_left = height[i - 1];
if (max_right == height[i]) {
// 重新寻找最大右边界
max_right=0;
for (int j = i + 1; j < height.size(); j++) {
if (height[j] > max_right)
max_right = height[j];
}
}
limit = min(max_left, max_right);
if (limit > height[i])
vol = vol + (limit - height[i]);
}
return vol;
}
};