LeetCode OJ 503. Next Greater Element II

最新推荐文章于 2026-06-26 23:12:21 发布

原创最新推荐文章于 2026-06-26 23:12:21 发布 · 273 阅读

0 ·

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

标签

#leetcode #stack

LeetCode 同时被 2 个专栏收录

22 篇文章

订阅专栏

stack

3 篇文章

订阅专栏

本文介绍了解决LeetCode OJ 503题的方法，题目要求寻找环形数组中每个元素的下一个更大元素。通过两次遍历数组形成闭环查找，并利用栈保存数组下标，实现高效求解。

LeetCode OJ 503. Next Greater Element II

Description

Given a circular array (the next element of the last element is the first element of the array), print the Next Greater Number for every element. The Next Greater Number of a number x is the first greater number to its traversing-order next in the array, which means you could search circularly to find its next greater number. If it doesn’t exist, output -1 for this number.

Example 1

Input: [1,2,1]
Output: [2,-1,2]
Explanation:
The first 1’s next greater number is 2;
The number 2 can’t find next greater number;
The second 1’s next greater number needs to search circularly, which is also 2.

Note:

The length of given array won’t exceed 10000.

解题思路

1、由于数组是形成一个环的，所以遍历数组两次，来形成环的查找。
2、使用stack：stack保存的是nums中数的下标。设栈顶是第k个数，对于nums中的第i个数，如果这个数大于stack的栈顶元素nums[k]，则nums[i]是大于nums[k]的第一个数。则结果数组ans[k]=nums[i]。一直到stack当前所有小于nums[i]的元素都比较完，把nums[i]存到stack中。循环以上操作直到数组遍历完。（因为我们遍历的是两个数组的长度，所以对于i要做模操作，保证结果数组大小等于原数组大小：i = i % nums.size()）

代码

个人github代码链接


class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> ans(nums.size(), -1);
        stack<int> s;
        for(int i = 0; i < 2 * nums.size(); i++){
            int num = nums[i % nums.size()];
            while(!s.empty() && nums[s.top()] < num){
                ans[s.top()] = num;
                s.pop();            
            }
            if(i < nums.size())   s.push(i);
        }
        return ans;
    }
};