LeetCode单调栈503|42

Erin Wang

于 2024-03-29 00:08:46 发布

阅读量200

点赞数 2

分类专栏： LeetCode 文章标签： leetcode 算法

本文链接：https://blog.youkuaiyun.com/ErinWang666/article/details/137127725

版权

LeetCode 专栏收录该内容

18 篇文章

订阅专栏

文章介绍了两个编程问题的解决方案：一个是寻找给定数组中每个元素右侧的第一个大于它的元素（nextGreaterElements），另一个是计算给定高度数组中可以接雨水的最大面积（trap）。两种问题都涉及使用堆栈数据结构进行高效查找和计算。

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

503下一个更大元素II

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

42接雨水

class Solution:
    def trap(self, height: List[int]) -> int:
        result = 0
        stack = [0]
        for i in range(1, len(height)):
            if height[i] < height[stack[-1]]:
                stack.append(i)
            elif height[i] == height[stack[-1]]:
                stack.pop()
                stack.append(i)
            else:
                while stack and height[i] > height[stack[-1]]:
                    mid = height[stack[-1]]
                    stack.pop()
                    if stack:
                        height_left = height[stack[-1]]
                        h = min(height_left, height[i]) - mid
                        w = i - stack[-1] - 1
                        result += h * w
                stack.append(i)
        return result