LeetCode-503. 下一个更大元素 II (单调栈)

系统管理员 2022-02-19 04:25 378阅读 0赞

# [503. 下一个更大元素 II][503. _ II] #

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

### 示例 1: ###

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

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

### 单调栈 ###

相比于前一个题，这道题相当于每个元素在数组查询两遍，让数组变成一个环，可以这样取下标`i%length`，将数组长度取原来的二倍，这样可以实现每个元素查询两遍的效果。  
哎吆，当时一直让栈直接存的数组中的元素，结果还有负数，后来想想可以直接存它的下标啊，这样还能少开一个数组呢！！

class Solution { 
    public:
        vector<int> nextGreaterElements(vector<int>& nums) { 
            vector<int> vec(nums.size(),-1);
            stack<int> st;  
            int len = nums.size();
            for(int i = 0; i < 2*len; i++)
            { 
                while( !st.empty() && nums[st.top()] < nums[i%len])
                { 
                    vec[st.top()] = nums[i%len];
                    st.pop();
                }
                if(i<len) // 查完第一遍后不再加入栈中，第二遍只是查找栈中剩余元素的目标解
                    st.push(i%len);
            }
            return vec;
        }
    };

[503. _ II]: https://leetcode-cn.com/problems/next-greater-element-ii/