1.题目如下:
给定 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
提示:
n == height.length
1 <= n <= 2 * 104
0 <= height[i] <= 105
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.代码如下:
class Solution {
public:
/*
* 思路:遍历每一个柱子的高度时,计算每个柱子左右两边的最高柱子
* 再然后由左右最高的柱子中较小的那个柱子减去所遍历的柱子高度的总和就是总储水量
* 单独计算每一个柱子的储水量相比于直接计算区间的储水量更直白简单,如果计算区间的储水量会有很多种情况,不容易划分
*
*/
int trap(vector<int>& height) {
int sum=0;
if(height.size()==0){
return 0;
}
int leftMax[height.size()];
int rightMax[height.size()];
leftMax[0]=height[0];
rightMax[height.size()-1]=height[height.size()-1];
for(int i=1;i<height.size();i++){
leftMax[i]=max(leftMax[i-1],height[i]);
}
for(int i=height.size()-2;i>=0;i--){
rightMax[i]=max(rightMax[i+1],height[i]);
}
for(int i=0;i<height.size();i++){
sum+=min(rightMax[i],leftMax[i])-height[i];
}
return sum;
}
};