leetcode下一个更大元素II

leetcode下一个更大元素II_第1张图片

1.暴力法

对于每一个元素,寻找比其大的第一个元素

需要注意的是,该数组是循环数组,对当前元素nums[i],首先在其后寻找是否存在比其大的元素,如果找不到,在从数组第一个元素开始寻找

代码如下:

class Solution {
public:
    vector nextGreaterElements(vector& nums) {
     //数组是循环数组
      int len=nums.size();
      vector res(len,-1);
      if(len==0)
         return res;
      for(int i=0;inums[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]

对于最后一个元素,从栈顶到栈底依次存放的是下标为0---nums.size()-2的元素,这样对于最后一个元素,就是从头开始往后判断是否存在一个值大于它;

手动模拟如下:如1 3 4 2

初始栈中为:

1

3

4  

对于2首先与1比较,弹出1,然后与3比较,成立,则res[3]=3,将2入栈,此时栈中元素为:

2

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 nextGreaterElements(vector& nums) {
        stack s;
        vector 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;
    }
};

 

 

 

 

 

 

 

 

 

 

 

你可能感兴趣的:(leetcode下一个更大元素II)