一、题目描述
给定 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
如果当前值i比max大,更新max;
如果当前值i比max小,看后面是否有大于等于max的数组值:
如果有,水量+=max-i;
如果没有,更新max为后面的最大值(要比i大),水量+=max-i;
如果没有比i大的,更新max为i;
如果相等,跳过;
- 关键点:找出从数组某一值开始到数组末尾的最大值
- 找最大值:使用vector存储,vector下标对应height下标,vector值为从下标的下一个位置到末尾的最大高度;从后向前遍历height,最大高度为max(下标+1的vector值,下标+1的height值)
时间复杂度:一次遍历找最大高度,一次遍历算水量;O(n)
空间复杂度:数组保存右边的最大高度;O(n)
四、代码
class Solution {
public:
int trap(vector<int>& height) {
//计算、保存右边的最大高度
vector<int>h(height.size(),0);
for(int i=height.size()-2;i>=0;i--)h[i]=max(h[i+1],height[i+1]);
int max=height[0];//左边的最大高度
int ans=0;//接水量
for(int i=1;i<height.size()-1;i++){
//若比左最大值大 或 比右最大值大,更新最大值
if(height[i]>=max||h[i]<height[i])max=height[i];
else{
//左最大值和右最大值中的较小值
int j=min(max,h[i]);
//接水量
ans+=j-height[i];
}
}
return ans;
}
};