【重点】【单调栈】503.下一个更大元素 II

原创已于 2025-06-18 11:44:55 修改 · 464 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#单调栈 #循环数组

于 2024-01-04 01:56:58 首次发布

labuladong的算法小抄笔记专栏收录该内容

85 篇文章

订阅专栏

文章介绍了如何使用单调栈数据结构解决编程题目，通过遍历循环数组并维护栈中的元素，找到每个位置的下一个更大元素。

题目

法1：单调栈+循环数组

参考思路：本题解答

Python

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

        return res

Java

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int n = nums.length;
        int[] res = new int[n];
        Stack<Integer> stack = new Stack<>();
        for (int i = 2 * n - 1; i >= 0; --i) { // 循环数组处理!!!
            while (!stack.isEmpty() && nums[i % n] >= stack.peek()) {
                stack.pop();
            }
            res[i % n] = stack.isEmpty() ? -1 : stack.peek();
            stack.push(nums[i % n]);
        }

        return res;
    }
}