Leetcode | 0503.下一个更大元素（2）

题目

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

思路

对数组进行遍历，当遇到单调递减的序列时，则它们的“下一个更大的元素”是相同的，则将这些数保存起来，直到找到一个比它们更大的元素。这个更大的元素如果比保存下来的这些数字都要大，则这个更大的元素就是这些数字的“下一个更大元素”。
实现过程使用“单调栈”，单调栈是说栈里面的元素从栈底到栈顶是单调递增或者单调递减的。本题使用单调递减栈，对原数组进行遍历：
1、如果栈为空，则把当前元素放入栈内。
2、如果栈不为空，则将当前遍历到的元素和栈顶的元素进行比较：
①如果栈顶的元素大于当前遍历到的元素：当前遍历到的元素的“下一个更大元素”是和栈顶元素的”下一个更大元素“一样的，于是将当前遍历到的元素入栈。
②如果栈顶的元素小于当前遍历到的元素：说明当前遍历到的元素是栈中一些元素的“下一个更大元素”，则逐个弹出栈顶元素，直到当前元素比栈顶元素小为止。出栈元素的”下一个更大元素”为当前遍历到的元素。

实现过程如下图所示：

代码实现

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