0
点赞
收藏
分享

微信扫一扫

503. 下一个更大元素 II

凶猛的小白兔 2022-01-20 阅读 43

要求:每个元素的下一个更大元素距离,可以循环
思路:单调栈,数组拉长为2n-1,即倒数第二个元素,即i<2n-1

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

相关推荐

0 条评论