1.题目描述
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
注意: 输入数组的长度不会超过 10000。
2.思路(单调栈)
我们首先把第一个元素 A[1] 放入栈,随后对于第二个元素 A[2],如果 A[2] > A[1],那么我们就找到了 A[1] 的下一个更大元素 A[2],此时就可以把 A[1] 出栈并把 A[2] 入栈;如果 A[2] <= A[1],我们就仅把 A[2] 入栈。对于第三个元素 A[3],此时栈中有若干个元素,那么所有比 A[3] 小的元素都找到了下一个更大元素(即 A[3]),因此可以出栈,在这之后,我们将 A[3] 入栈,以此类推。
3.代码
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int n = nums.size();
vector<int> next(n, -1);
stack<int> pre;
for(int i = 0;i < 2*n; ++i){
int num = nums[i % n];
while(!pre.empty() && nums[pre.top()] < num){
next[pre.top()] = num;
pre.pop();
}
if(i < n){
pre.push(i);
}
}
return next;
}
};
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(n)