栈和队列 - 循环数组中比当前元素大的下一个元素

最新推荐文章于 2025-07-20 23:25:32 发布

原创最新推荐文章于 2025-07-20 23:25:32 发布 · 381 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #数据结构 #算法 #动态规划

LeetCode 专栏收录该内容

78 篇文章

订阅专栏

这篇博客介绍了如何利用单调栈解决LeetCode中503题——下一个更大元素II的问题。通过遍历数组两次，用栈来存储当前遍历到的元素索引，当遇到比栈顶元素大的数时，更新栈顶元素对应的下一个更大元素，最后返回结果数组。时间复杂度为O(n)，空间复杂度为O(n)。

1.题目

LeetCode: 503. 下一个更大元素 II

【medium】

给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。

示例 1:

输入: [1,2,1]

输出: [2,-1,2]

解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数；
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

注意:
输入数组的长度不会超过 10000。

2.解题

方法一：单调栈

此题与LeetCode: 739. 每日温度很相似，区别在于循环数组，因此遍历数组时要遍历两遍，如果没有更大的则为-1，所以将数组初始化为-1。

java：

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int n = nums.length;
        int res = new int[n];
        Arrays.fill(res, -1);
        Stack<Integer> s = new Stack<>();
        for (int i = 0; i < n * 2; i++) {
            int num = nums[i % n];
            while (!s.isEmpty() && num > nums[s.peek()]) {
                res[s.pop()] = num;
            }
            if (i < n) s.push(i);
        }
        return res;
    }
}