题目描述:
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 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
思路:
每个位置,统计左边最高的高度,和右边的最高的高度
然后再减去当前位置的高度,就是本位置下能接的所有雨水
参考:https://leetcode-cn.com/problems/trapping-rain-water/comments/
代码如下:
class Solution {
public:
int trap(vector<int>& height) {
int n=height.size();
vector<int>left(n,0),right(n,0);
for(int i=1;i<n;i++){
left[i]=max(left[i-1],height[i-1]);
}
for(int i=n-2;i>=0;i--){
right[i]=max(right[i+1],height[i+1]);
}
int res=0;
for(int i=0;i<n;i++){
int level=min(left[i],right[i]);
res+=max(0,level-height[i]);
}
return res;
}
};