leetcode刷题记录 2018.7.6

最新推荐文章于 2025-01-14 21:25:25 发布

super涵

最新推荐文章于 2025-01-14 21:25:25 发布

阅读量185

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/qq_23511015/article/details/80945957

leetcode 专栏收录该内容

7 篇文章

订阅专栏

本文针对经典的接雨水问题，提供了一种高效求解方法。通过找到数组中的最高柱子，并从两侧向最高点逼近，动态更新次高柱子高度，以此来计算能够存储的雨水总量。

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

1.接雨水问题

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

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。感谢 Marcos 贡献此图。

示例:

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

分析：根据题目中的要求，首先想到的是，要能存住水，一定要满足两边的比中间高。所以我们可以先找出最高的，然后从两边向最高的靠近，定义一个变量记录两边的次高值，当次高值的下一个值比次高值小，肯定可以存住水，如果发现了更高的值，就更新次高值，直到逼近最高值结束。由此可以写程序

class Solution:
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        second = 0
        result = 0
        maxid = 0
        for i in range(1, len(height)):
            if height[i] > height[maxid]:
                maxid = i
        for i in range(0, maxid):
            if second < height[i]:
                second = height[i]
            else:
                result += second - height[i]
        second = 0
 
        for j in range(len(height) - 1, maxid, -1):
            if second < height[j]:
                second = height[j]
            else:
                result += second - height[j]
        return result