描述:
给定 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
思路:
把柱子和接的雨水看成一个整体,头尾相向指针,则雨水高度取决左右边界最短的一块板,左边界小于等于右边界则丢弃短板,反正亦然;分别两个计数器,通过淘汰边界指针记下整体高度和及柱子高度和,差值即为雨水的容量,最后均需加上最高的柱子高度,又因计数器后面相减则可以考虑不用计算最后的柱子。
题解:
class Solution {
public int trap(int[] height) {
int ans = 0;
int left = 0, right = height.length - 1;
int hi = 0;
int ans2 = 0;
while (left < right){
if (height[left] < height[right]){
hi = Math.max(hi, height[left]);
ans += hi;
ans2 += height[left];
left++;
}
else {
hi = Math.max(hi, height[right]);
ans += hi;
ans2 += height[right];
right--;
}
}
return ans - ans2;
}
}
复杂度分析
时间复杂度:O(n),其中 n 是数组 height 的长度。两个指针的移动总次数不超过 n。
空间复杂度:O(1)。只需要使用常数的额外空间。
作者:力扣官方题解
原题链接:42.接雨水
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。