给定 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
思路:双指针,第一个指针指向第一根柱子,第二个指针从第二根柱子往后遍历,直到找到一根比第一个指针指向的柱子高或者等高的柱子。然后计算空位,用矮的柱子高度乘以他们两个之间的距离减一,然后减去他们中间的其它更矮的柱子占的空间(用一个循环来遍历他们之间的矮柱子),就是接的雨水。然后第一个指针改变指向,指向第二跟柱子的位置,第二个指针再往后继续上述流程。
最后,第一个指针会停留在最高的柱子上。
通过上述例子的图可以看到我们目前之完成了一半的雨水计算,即最高的柱子的左边接的雨水。现在我们可以按照同样的思路,从后往前遍历,就能得到最高柱子右边接的雨水。起点是数字最后一位元素,终点可以设为最高柱子处,不用跑到0,节省了时间。
int trap(int* height, int heightSize) {
int count=0;
int left=0;
for(int i=1;i<heightSize;i++)//正向找
{
if(height[i]<height[left])
{
continue;
}
else
{
if(i-left>1)//两个柱子之间有间隙再计算,没有间隙直接跳过
{count+=height[left]*(i-left-1);//用矮的柱子来计算总的格子
for(int j=left+1;j<i;j++)
{
count-=height[j];//减去两个柱子中间的矮柱子占的空间
}
}
}
left=i;//跳到下一根高柱子
}
int right=heightSize-1;
for(int i=heightSize-2;i>=left;i--)//反向找
{
if(height[i]<height[right])
{
continue;
}
else
{
if(right-i>1)
{count+=height[right]*(right-i-1);
for(int j=i+1;j<right;j++)
{
count-=height[j];
}
}
right=i;
}
}
return count;
}