70-Stack--LC503下一个更大元素II

最新推荐文章于 2025-12-10 14:25:37 发布

原创最新推荐文章于 2025-12-10 14:25:37 发布 · 136 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

75 篇文章

订阅专栏

这篇博客介绍了如何使用单调栈和哈希表解决数组中每个元素的下一个更大元素问题，特别关注了处理循环数组的方法，通过实例代码展示了如何利用下标取模实现。

在这里插入图片描述

class Solution(object):
    def nextGreaterElements(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        # 单调栈+哈希表，跟LC496的区别是，这次是循环数组
        # 比较笨的办法就是在给出的数组后面接上数组的[0,n-1]，这样就算人工构造的循环数组了
        # 其实，可以直接用下标取模来实现
        n = len(nums)
        res = [-1] * n
        stack = list()
        for i in range(2*n-1):
            while stack and nums[i%n] > nums[stack[-1]]:
                res[stack.pop()] = nums[i%n]
            stack.append(i%n)
        return res
        # 逆序
        n = len(nums)
        res = [-1]*n
        stack = list()
        for i in range(2*n-1, -1, -1):
            while stack and nums[stack[-1]] <= nums[i%n]:
                stack.pop()
            res[i%n] = nums[stack[-1]] if stack else -1
            stack.append(i%n)
        return res