接雨水
给定 n 个非负整数表示每个宽度为 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 个单位的雨水(蓝色部分表示雨水)。
1
2
3
2
3
/**
* @param {number[]} height
* @return {number}
*/
var trap = function(height) {
let ans = 0;
let left = 0, right = height.length - 1;
let leftMax = 0, rightMax = 0;
while (left < right) {
// 每次指针移动后,更新左右两侧的最大高度
leftMax = Math.max(leftMax, height[left]);
rightMax = Math.max(rightMax, height[right]);
// 如果左侧的点小于右侧的点,那么左侧点的接水量其实只取决于左侧的最大高度和本身的高度了
if (height[left] < height[right]) {
ans += leftMax - height[left];
++left;
} else {
// 右侧同理
ans += rightMax - height[right];
--right;
}
}
return ans
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/trapping-rain-water