LeetCode42.接雨水 JavaScript

给定 n 个非负整数表示每一个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨以后能接多少雨水。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种状况下,能够接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。javascript

示例:java

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

答案参考:数组

/**
 * @param {number[]} height
 * @return {number}
 */
var trap = function (height) {
    let left = 0, right = height.length - 1
    let count = 0
    let leftMax = 0, rightMax = 0
    while (left <= right) {
        leftMax = Math.max(leftMax, height[left])
        rightMax = Math.max(rightMax, height[right])
        if (leftMax < rightMax) {
            count += leftMax - height[left]
            left++
        } else {
            count += rightMax - height[right]
            right--
        }
    }
    return count
};

欢迎关注

相关文章
相关标签/搜索