LeetCode 496. 下一个更大元素 I 解题思路及C++实现

解题思路：

因为这两个int型的vector中不包含重复元素，所以考虑使用c++中的unordered_map，对num2进行遍历，找到每一个元素的下一个更大元素，这样的话，时间复杂度就是O(n)。

如果使用暴力的方法：对nums1中的每个int，在nums2中去找下一个更大元素，时间复杂度将会是O(num1.size() * nums2.size())，太过暴力。

使用一个unordered_map和一个stack进行处理的方法如下：

• 大循环遍历vector nums2；
• 当栈不为空且栈顶元素小于nums2当前元素时，说明找到了栈顶元素的下一个更大元素，将其记录在unordered_map中；
• 如果栈为空，或者栈顶元素仍大于当前的nums2元素，则将当前的nums2元素入栈。

循环下来之后，nums2中的一些元素可能会没有相应的下一个更大元素，留在了栈中，但我们不管这个了，不用再将其对应-1存进unordered_map中（为了省空间）。而且，最终结果可以直接在nums1中进行更新（省空间）。

 

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int, int> a;
        stack<int> b;
        
        //遍历nums2
        for(auto x: nums2){
            while(!b.empty() && b.top() < x){
                a[b.top()] = x;
                b.pop();
            }
            b.push(x);
        }
        
        //将结果直接在nums1中进行更新
        for(int i = 0; i < nums1.size(); i++){
            if(a.count(nums1[i]))  //利用count函数来看看unordered_map中是否有这个key，有的话返回1，无 则返回0
               nums1[i] = (a[nums1[i]]);
            else nums1[i] = -1;
        }
        return nums1;
    }
};