59 第九章单调栈

最新推荐文章于 2024-07-20 14:12:50 发布

qq_37420393

最新推荐文章于 2024-07-20 14:12:50 发布

阅读量70

点赞数

分类专栏：代码随想训练营文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_37420393/article/details/128974899

版权

代码随想训练营专栏收录该内容

50 篇文章 0 订阅

订阅专栏

59 第九章单调栈
● 503.下一个更大元素II
● 42. 接雨水
在这里插入图片描述

class Solution:
    def trap(self, height: List[int]) -> int:
        # 42. 接雨水
        q = collections.deque()
        res = height.copy()
        preheght_idx = -1        
        for i in range(len(height)):
            # 寻找右边第一个比现在大的值，一直找到最左边，记录左边起始值
            b = True
            while q and height[i] >= height[q[-1]]:
                b = False
                preheght_idx = q[-1]
                q.pop()
            q.append(i)
            if b:
                continue
            # 将这段洼地置为左边填充后水平面的高度
            for r in range(preheght_idx, i):
                res[r] = max(height[preheght_idx], res[r])

        if len(q) >= 2:
            for i in range(len(q)-1, 0, -1):
                for j in range(q[i-1]+1, q[i]):
                    res[j] = max(res[j], height[q[i]])
        re = 0
        for i in range(len(height)):
            re += (res[i] - height[i])
        return re