题目
42. 接雨水
给定 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 个单位的雨水(蓝色部分表示雨水)。
动态规划
max_left[]:当前位置i左侧最高的柱子
max_right[]:当前位置i右侧最高的柱子
C++代码
class Solution {
public:
int trap(vector<int>& height) {
int sum=0;
int n =height.size();
vector<int> max_left(n);
vector<int>max_right(n);
for(int i=1;i<n-1;i++)
max_left[i]=max(max_left[i-1],height[i-1]);
for(int i=n-2;i>=0;i--) //从右往前求
max_right[i]=max(max_right[i+1],height[i+1]);
for(int i=1;i<n-1;i++)
{
int minn=min(max_left[i],max_right[i]);
if(minn>height[i])
sum+=minn-height[i];
}
return sum;
}
};