leetcode下一个更大元素I

1.暴力法

遍历num1中的每一个元素，在nums2中寻找比其大的元素

代码如下：

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
      vector<int> res;
      int len=nums2.size();
      for(int i=0;i<nums1.size();i++)
    {
      int j=0;
      for(;j<len;j++)
        if(nums2[j]==nums1[i])//nums2中寻找与其相等的元素
          break;
      for(;j<len;j++)
        if(nums2[j]>nums1[i])//判断是否存在比其大的元素
          break;
        if(j==len)
          res.push_back(-1);
        else
          res.push_back(nums2[j]);
    }
    return res;
    }
};

2.栈+map

对于nums2中的每一个元素nums2[i]:

  若栈非空且大于栈顶元素，则将栈顶元素弹出，并(栈顶元素，nums2[i])的形式存入map,直到栈顶元素大于等于Nums2[i]

然后将当前元素入栈，参与寻找比其大的元素。

可以手动模拟一下

代码如下：

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
      vector<int> res;
      int len=nums2.size();
      stack<int> s;
      map<int,int> m;
      for(int i=0;i<len;i++)
      {
             while(!s.empty()&&nums2[i]>s.top())
             {
               m[s.top()]=nums2[i];
               s.pop();
             }
              s.push(nums2[i]);
      }
      //此时栈中保存的都是未找到比其大的元素
      while(!s.empty())
      {
          m[s.top()]=-1;
          s.pop();
      }
      for(int i=0;i<nums1.size();i++)
           res.push_back(m[nums1[i]]);
    return res;
    }
};