给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:第i个单位高度为0时能装下的水等于,(从左边看的最大值,从右边看的最大值)的最小值。
有高度时候对应减小即可。
空间复杂度O(n),时间复杂度O(n)
class Solution {
public:
int trap(vector<int>& height) {
vector<int> maxl;
vector<int> maxr;
for(int i=0;i<height.size();i++){
if(maxl.size()==0){
maxl.push_back(height[i]);
}else if(maxl[maxl.size()-1]<height[i]){
maxl.push_back(height[i]);
}else{
maxl.push_back(maxl[maxl.size()-1]);
}
}
for(int i=height.size()-1;i>=0;i--){
if(maxr.size()==0){
maxr.push_back(height[i]);
}else if(maxr[maxr.size()-1]<height[i]){
maxr.push_back(height[i]);
}else{
maxr.push_back(maxr[maxr.size()-1]);
}
}
reverse(maxr.begin(),maxr.end());
int sum = 0;
for(int i=0;i<maxl.size();i++){
int w = min(maxl[i],maxr[i]);
int l = max(w-height[i],0);
sum+= l;
}
return sum;
}
};