1.暴力法
对于每一个元素,寻找比其大的第一个元素
需要注意的是,该数组是循环数组,对当前元素nums[i],首先在其后寻找是否存在比其大的元素,如果找不到,在从数组第一个元素开始寻找
代码如下:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
//数组是循环数组
int len=nums.size();
vector<int> res(len,-1);
if(len==0)
return res;
for(int i=0;i<len;i++)
{
int j=(i+1)%len;//因为是循环数组
while(j!=i)
{
if(nums[j]>nums[i])
break;
j=(j+1)%len;
}
if(j==i)
res[i]=-1;
else
res[i]=nums[j];
}
return res;
}
};
2.栈
需要注意的是是循环数组
首先逆序入栈,数组最后一个元素不入栈,
然后从后往前遍历,对于当前元素nums[i],判断nums[i]>=s.top是否成立,如果成立,则弹出栈顶元素,直到栈为空或者存在一个栈顶元素大于nums[i]。然后将当前元素入栈。
对于nums[i-1],首先与栈顶元素即num[i]比较,如果nums[i-1]<nums[i],则结果为nums[i];如果不成立,则弹出栈顶元素(在上一轮中,比nums[i]小的已经从栈中弹出)继续比较,
对于最后一个元素,从栈顶到栈底依次存放的是下标为0---nums.size()-2的元素,这样对于最后一个元素,就是从头开始往后判断是否存在一个值大于它;
手动模拟如下:如1 3 4 2
初始栈中为:
1
3
4
对于2首先与1比较,弹出1,然后与3比较,成立,则res[3]=3,将2入栈,此时栈中元素为:
2
3
4
对于4,与2比较,不成立,弹出2(由于1<2,上一轮已经被弹出,且2<3),然后与3比较,弹出3,然后与4比较,弹出4,此时栈为空,说明找不到,则res[2]=-1,将4入栈,此时栈中元素为:
4
对于3,与栈顶4比较,成立,则res[1]=4,将3入栈,此时栈为
3
4
对于1,与栈顶3比较,成立,则res[0]=3,将1入栈,结束遍历
代码如下:
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
stack<int> s;
vector<int> res(nums.size());
for(int i=nums.size()-2;i>=0;i--) s.push(nums[i]);
for(int i=nums.size()-1;i>=0;i--){
while(!s.empty()&&nums[i]>=s.top())
s.pop();
res[i]=s.empty()?-1:s.top();
s.push(nums[i]);
}
return res;
}
};