代码随想录算法训练营第五十一天|503.下一个更大元素II，42. 接雨水

最新推荐文章于 2024-06-04 22:50:27 发布

丁希希哇

最新推荐文章于 2024-06-04 22:50:27 发布

阅读量832

点赞数 16

分类专栏：力扣算法刷题文章标签：算法面试 python 力扣

本文链接：https://blog.csdn.net/weixin_47748259/article/details/136676739

版权

力扣算法刷题专栏收录该内容

69 篇文章 3 订阅

订阅专栏

系列文章目录

文章目录

系列文章目录
503.下一个更大元素II
42. 接雨水

503.下一个更大元素II

题目链接： 503.下一个更大元素II
题目内容： 给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1 。
视频讲解： 单调栈，成环了可怎么办？LeetCode：503.下一个更大元素II

这道题和496.下一个更大元素 I的区别在于可以循环的对数组进行搜索

思路：将两个nums数组拼接在一起，使用单调栈计算出每一个元素的下一个最大值，最后再把结果集即result数组resize到原数组大小。

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        dp=[-1]*len(nums)
        stack=[]
        for i in range(len(nums)*2):
            while (len(stack)!=0 and nums[i%len(nums)]>nums[stack[-1]]):
                dp[stack[-1]]=nums[i%len(nums)]
                stack.pop()
            stack.append(i%len(nums))
        return dp

42. 接雨水

题目链接： 42. 接雨水
题目内容： 给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
视频讲解： 单调栈，经典来袭！LeetCode:42.接雨水

class Solution:
    def trap(self, height: List[int]) -> int:
        stack = [0]
        result = 0
        for i in range(1, len(height)):
            while stack and height[i] > height[stack[-1]]:
                mid_height = stack.pop()
                if stack:
                    # 雨水高度是 min(凹槽左侧高度, 凹槽右侧高度) - 凹槽底部高度
                    h = min(height[stack[-1]], height[i]) - height[mid_height]
                    # 雨水宽度是 凹槽右侧的下标 - 凹槽左侧的下标 - 1
                    w = i - stack[-1] - 1
                    # 累计总雨水体积
                    result += h * w
            stack.append(i)
        return result

丁希希哇

关注

16
点赞
踩
21

收藏

觉得还不错? 一键收藏
0
评论
代码随想录算法训练营第五十一天|503.下一个更大元素II，42. 接雨水

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
复制链接

扫一扫