＜leetcode＞503.下一更大元素——单调栈

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。

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

来源：力扣（LeetCode）

解答：

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        
        vector<int> ans;
        for(int i = 0; i < size(nums); i++){
            int res = -1;
            for(int j = i + 1;j < size(nums); j++){
                if(nums[j] > nums[i]){
                    ans.push_back(nums[j]);
                    res = 0;
                    break;
                }
            }
            if(res < 0){
                for(int j = 0; j < i; j++){
                    if(nums[j] > nums[i]){
                        ans.push_back(nums[j]);
                        res = 0;
                        break;
                    }

                }
            }
            if(res < 0) ans.push_back(-1);
        }
        return ans;
    }
};

优化：

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