给定 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.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
可以将该问题看作是左右两个木板之间的接水问题
首先使用 left 和 right 作为当前遍历所得到的最左和最右的最大值,比较 left 和 right 的大小,谁小就谁移动(因为水的体积大小是由最小的木板决定的)。
在移动过程中,left 或者 right 如果遇见更大的值就更新 left 或者 right,这时候就是一个新的两木板之间的水的体积问题。
在 left 和 right 中较小木板移动的过程中,遇见小的木板,其体积可以表达为:left(或 right)减去 (当前木板高度)。
代码如下(示例):
class Solution {
public:
int trap(vector<int>& height) {
//获得当前左边的第一个值,作为初始的左边最大值
int left=height[0];
//获得当前右边的第一个值,作为初始的右边最大值
int right=height[height.size()-1];
//初始化当前体积(或者是面积,这里我叫做面积)
int area=0;
//循环在从左和从右遍历两个坐标,i 和 j 相等时结束
for(int i=1,j=height.size()-2;i<=j;)
{
//如果 left 小于 right,检查是否更新面积 area 和 left,左坐标 i 移动
if(left<=right)
{
// left 比当前值大,不更新 left,area 加上当前面积,左坐标 i 移动
if(left>height[i])
{
area+=(left-height[i]);
i++;
}
// left 比当前值小,更新 left,area不增加,左坐标 i 移动
else
{
left=height[i];
i++;
}
}
//如果 left 大于 right,检查是否更新面积 area 和 right,右坐标 j 移动
else
{
// right 比当前值大,不更新 right,area 加上当前面积,右坐标 j 移动
if(right>height[j])
{
area+=(right-height[j]);
j--;
}
// right 比当前值小,更新 right,area不增加,右坐标 j 移动
else
{
right=height[j];
j--;
}
}
}
return area;
}
};