leetcode 42. 接雨水

最新推荐文章于 2024-07-08 16:31:27 发布

原创最新推荐文章于 2024-07-08 16:31:27 发布 · 180 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

9 篇文章

订阅专栏

该博客介绍了一个计算数组表示的高度图中雨水容量的问题，通过动态规划的方法，从左右两侧分别计算最大高度，然后计算每个位置能接的雨水量，最终得出总雨水量。示例给出了一个具体的输入和输出解释。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定 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 个单位的雨水（蓝色部分表示雨水）。

/**
 * @param {number[]} height
 * @return {number}
 */
//使用动态规划
var trap = function (height) {
    let n = height.length
    if (n < 2) return 0
    //从右边计算最大数组
    const leftMax = new Array(n).fill(0)
    leftMax[0] = height[0]
    for (let i = 1; i < n; i++) {
        leftMax[i] = Math.max(leftMax[i - 1], height[i])
    }
    //从右边计算最大数组
    const rightMax = new Array(n).fill(0)
    rightMax[n - 1] = height[n - 1]
    for (let j = n - 2; j >= 0; j--) {
        rightMax[j] = Math.max(rightMax[j + 1], height[j])
    }
    //计算雨水总单位数
    let ans = 0
    for (let s = 0; s < n; s++) {
        ans += Math.min(leftMax[s], rightMax[s]) - height[s]
    }
    return ans
};