题目:给定 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解决方案,考虑计算每一列能够存储的雨水数量,第一列和最后一列是无法存储的,故不考虑这两列,其次,每一列能够存水的数量就等于它左右两边最高的两列中,较小的一列高度与本列的差值,
例如,图片中第三列本身高度为0,左边最高值为1,右边最高值为三,它能存储的水量就是1-0=1;
图片第五列高度为1,其左方最高值为2,右方最高值为3,求差为1,循环求出第2到n-1列的存储量,并求和,即为最终答案。
解决代码如下
class Solution {
public int trap(int[] height)
{ int num =0;
if (height.length==0)
{
return 0;
}
for(int i=1;i<height.length;i++)
{
int max[]=new int[2];
for(int j=i;j>=0;j--)
{
if (max[0]<height[j])
{
max[0]=height[j];
}
}
for(int j=i;j<=height.length-1;j++)
{
if(max[1]<height[j])
{
max[1]=height[j];
}
}
int w = Math.min(max[0],max[1])-height[i];
if(w>0)
{
num+=w;
}
}
return num;