1、描述
503给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-greater-element-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2、关键字
下一个更大更小的数
3、思路
单调栈,
我们将当前还没得到答案的下标暂存于栈内,从而实现「被动」更新答案。
思路二:双指针,后一个指针就是模拟往后循环遍历,去寻找结果:
通过「主动」遍历来实现的。
4、notes
寻找下一个更大更小,首选单调栈
5、复杂度:
单调栈:
时间:O(N)
空间:O(N)
双指针:
时间:O(n平方)
空间:O(N)
6、code
单调栈:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int n=nums.size();
vector<int>res(n,-1); // 结果集
stack<int>st;
for(int i=0;i<n*2-1;i++){ // 构造循环数组
while(!st.empty() && nums[i%n]>nums[st.top()]){ // 根据题意构造,单调递减栈,如果是当前数组i的结果,就把结果输出,被动寻找结果
res[st.top()]=nums[i%n];
st.pop();
}
st.push(i%n); // 存的是下标
}
return res;
}
};
循环双指针遍历:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int n=nums.size();
vector<int>res(n); // 结果集
for(int i=0;i<n;i++){ // 双指针往后遍历
int j=i;
while(++j){
if(j==n){ // 最后就跳到首
j=0;
}
if(i==j){ // 循环到自己了
res[i]=-1;
break;
}
if(nums[j]>nums[i]){ //找到最近的最大了
res[i]=nums[j];
break;
}
}
}
return res;
}
};