LeetCode - #42 接雨水（Top 100）

前言

本题为 LeetCode 前 100 高频题

我们社区陆续会将顾毅（Netflix 增长黑客，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新了 41 期，我们会保持更新时间和进度（周一、周三、周五早上 9:00 发布），每期的内容不多，我们希望大家可以在上班路上阅读，长久积累会有很大提升。

不积跬步，无以至千里；不积小流，无以成江海，Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言，我们会尽力满足大家的需求。

难度水平：困难

1. 描述

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

2. 示例

示例 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 * 10^4
• 0 <= height[i] <= 10^5

3. 答案

class TrappingRainWater {
   
    func trap(height: [Int]) -> Int {
   
        guard height.count > 0 else {
   
            return 0
        }

        var res = 0
        var left = _initMaxHeights(height, true)
        var right = _initMaxHeights(height, false)

        for i in 0..<height.count {
   
            res += min(left[i], right[i]) - height[i]
        }

        return res
    }

    private func _initMaxHeights(height: [Int], _ isLeft: Bool) -> [Int] {
   
        var res = [Int](count: height.count, repeatedValue: 0)
        var currentMax = 0

        if isLeft {
   
            for i in 0..<height.count {
   
                res[i] = max(currentMax, height[i])
                currentMax = res[i]
            }
        } else {
   
            for i in (0..<height.count).reverse() {
   
                res[i] = max(currentMax, height[i])
                currentMax = res[i]
            }
        }

        return res
    }
}

• 主要思想：雨容总是由左右最大高度的最小值决定的。
• 时间复杂度： O(n)
• 空间复杂度： O(n)

该算法题解的仓库：LeetCode-Swift

点击前往 LeetCode 练习

关于我们

我们是由 Swift 爱好者共同维护，我们会分享以 Swift 实战、SwiftUI、Swift 基础为核心的技术内容，也整理收集优秀的学习资料。

后续还会翻译大量资料到我们公众号，有感兴趣的朋友，可以加入我们。