题目
给定 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
思路:动态规划
1、沿Y轴按列计算,这样宽度为1,求面积只需要求每一列的雨水高度。
2、每一列雨水的高度等于左侧最高的柱子和右侧最高的柱子中高度最低值 减去 当前列所在柱子的自身高度,即为:
min(lHeight,rHeight) - height (第一列和最后一列不参与计算,无法接雨水)
3、事实上,遍历到一列时,完全可以以此为中心,向左右两边延伸寻找左侧最高和右侧最高的柱子,但是这样一来做了很多重复工作。
因此采用动态规划,从左到右遍历得到每一列的左边最大高度,定义为左边界 = 前一个左边最大高度与本高度的最大值。
从右到左遍历得到每一列的的右边最大高度,定义为右边界 = 为前一个左边最大高度与本高度的最大值。
左边界、柱子、右边界序号定义如图所示:
根据左边界右边界的定义依次求出柱子左侧、右侧的最大高度
4、有了左右边界,即可遍历所有柱子,按照步骤2中公式,将各柱子上方的雨水高度乘以宽度(1)即为雨水面积,依次累加得到总面积。
代码实现:
class Solution{
public:
int trap(vector<int>& height)
{
if(height.size() <= 2) return 0;
vector<int> maxLeft(height.size(),0);
vector<int> maxRight(height.size(),0);
int size = maxRight.size();
//记录每个柱子左边界:左边柱子最大高度
maxLeft[0] = height[0];
for(int i = 1; i < size;i++)
{
maxLeft[i] - max(height[i],maxLeft[i - 1]);
}
//记录每个柱子右边界:右边柱子最大高度
maxRight[size - 1] = height[size - 1];
for(int i = size - 2; i >= 0;i--)
{
maxRight[i] = max(height[i],maxRight[i + 1]);
}
//求和
int sum = 0;
for(int i = 0; i < size; i++)
{
int count = min(maxLeft[i], maxRight[i]) - height[i];
if(count > 0) sum += count;
}
return sum;
}
};