Python描述 LeetCode 42. 接雨水

Posted 2022-06-14 亓官劼

Python描述 LeetCode 42. 接雨水

  大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、CSDN、GitHub、B站等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文，时光荏苒，未来可期，加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言，我会尽快回复消息.

---

本文原创为【亓官劼】（qí guān jié ），请大家支持原创，部分平台一直在恶意盗取博主的文章！！！ 全部文章请关注微信公众号【亓官劼】。

题目

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

示例 2：

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

提示：

• n == height.length
• 1 <= n <= 2 * 104
• 0 <= height[i] <= 105

解题思路

分别找出从左到右的最大值，和从右到左的最大值。那么对于每一个height来说，它可以存储的雨水就是min(left_max,right_max)-height[i]，一次遍历就好

Python描述

class Solution:
    def trap(self, height: List[int]) -> int:
        left_max,right_max = [],[]
        max_height = 0
        for item in height:
            left_max.append(max(max_height,item))
            max_height = max(max_height,item)
        
        max_height = 0
        for item in height[::-1]:
            right_max.append(max(max_height,item))
            max_height = max(max_height,item)
        
        right_max = right_max[::-1]

        res = 0
        for i in range(len(height)):
            res += min(left_max[i],right_max[i]) - height[i]
        return res