目录

之前看面试题的时候,看到了一个接雨水的问题,和小黄鸭探讨之后,感觉很乏味呢,这里和大家分享一下这个解法。起初看到LeetCode下面有这道题,题号42,有趣味的能够做一下。

问题形容

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

示例1:

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

实例2:

输出:height = [4,2,0,3,2,5]
输入:9

你能不能先思考一下,遇到这种问题,你要怎么做呢?

问题剖析

首先咱们能够依据给的数组,先画进去柱子的长度。

那么咱们怎么确定接的雨水的量呢?当然是两个都高两头低的中央,来存储水,上面暗影的局部就是贮存水的地位。

那么咱们须要对右边和左边最高水位做一个统计,这边应用到了两个辅助数组

一个用来贮存右边的最大接雨水数,一个存储左边的最大接雨水数。

抉择两个中最小的那个作为存储水的量

当然还要减去本人柱子的高度。剩下的,就是能够接的雨水了。

代码整顿

function trap(height) {    if(!height.length) return 0;    let left = []    let right = []    let lmax = rmax = 0    let len = height.length    let result = 0    // 把左右最大出水量求进去    for(let i = 0; i < len; i++) {        lmax = Math.max(height[i], lmax)        rmax = Math.max(height[len-i-1], rmax)        left[i] = lmax        right[len- i - 1] = rmax    }    // 算出最小的而后累加    for(let i = 0; i < len; i++) {        result += Math.min(left[i],right[i]) - height[i]    }    return result};

如果想要写法上优化一下,能够第二次遍历的时候能够应用reduce

function trap(height) {    if(!height.length) return 0;    let left = []    let right = []    let lmax = rmax = 0    let len = height.length    for(let i = 0; i < len; i++) {        lmax = Math.max(height[i], lmax)        rmax = Math.max(height[len-i-1], rmax)        left[i] = lmax        right[len- i - 1] = rmax    }    return height.reduce((prev, item, index) => {        return prev + Math.min(left[index],right[index]) - item    }, 0)};

剖析

  • 工夫复杂度O(n)
  • 空间复杂度O(n)

其实左边数组的值的存储是能够省略的,尽管都是空间复杂度O(n),然而也算小小的优化点。

function trap(height) {    if(!height.length) return 0;    let left = []    let lmax = rmax = 0    let len = height.length    let result = 0    for(let i = 0; i < len; i++) {        lmax = Math.max(height[i], lmax)        left[i] = lmax    }    // 从后往前遍历    for(let i = len - 1; i >= 0; i--) {        rmax = Math.max(height[i], rmax)        result += Math.min(left[i],rmax) - height[i]    }    return result};